PACKAGE

IVORY.NNABLA

The ivory.nnabla.trainer module provides the Trainer class for nnable.

Classes
  • Trainer Trainer(epoch: int = -1, epochs: int = 1, global_step: int = -1, batch_size: int = 32, shuffle: bool = True, dataloaders: str = 'ivory.nnabla.data.DataLoaders', verbose: int = 1, loss: Union[Callable, NoneType] = None, gpu: bool = False, precision: int = 32, amp_level: str = 'O1')</>