toad.nn module

• toad.nn.module module
  • Module
    • Module.__init__()
    • Module.device
    • Module.fit()
    • Module.evaluate()
    • Module.fit_step()
    • Module.save()
    • Module.load()
    • Module.log()
    • Module.distributed()
  • DistModule
• toad.nn.functional module
  • flooding()
  • focal_loss()
  • binary_focal_loss()
  • focal_loss_for_numpy()
  • label_smoothing()
• toad.nn.trainer module
  • History
    • History.__init__()
    • History.log()
  • callback
    • callback.__init__()
  • earlystopping
    • earlystopping.setup()
    • earlystopping.get_best_state()
    • earlystopping.reset()
  • Trainer
    • Trainer.__init__()
    • Trainer.set_model()
    • Trainer.distributed()
    • Trainer.train()
    • Trainer.evaluate()