MODULE
IVORY.NNABLA.TRAINER
The ivory.nnabla.trainer
module provides the Trainer
class for nnable.
Classes
dataclass
ivory.nnabla.trainer.
Trainer
(
epoch=-1
, epochs=1
, global_step=-1
, batch_size=32
, shuffle=True
, dataloaders='ivory.nnabla.data.DataLoaders'
, verbose=1
, loss=None
, gpu=False
, precision=32
, amp_level='O1'
)
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')
Parameters
epoch
(int, optional) —epochs
(int, optional) —global_step
(int, optional) —batch_size
(int, optional) —shuffle
(bool, optional) —dataloaders
(str, optional) —verbose
(int, optional) —loss
(callable, optional) —gpu
(bool, optional) —precision
(int, optional) —amp_level
(str, optional) —
Attributes
amp_level
(str) —batch_size
(int) —dataloaders
(str) —epoch
(int) —epochs
(int) —global_step
(int) —gpu
(bool) —loss
(callable, optional) —precision
(int) —shuffle
(bool) —verbose
(int) —