Train Config¶
-
class
piepline.train_config.train_config.
BaseTrainConfig
(model: torch.nn.modules.module.Module, train_stages: [], loss: torch.nn.modules.module.Module, optimizer: torch.optim.optimizer.Optimizer)[source]¶ Train process setting storage
Parameters: - train_stages – list of stages for train loop
- loss – loss criterion
- optimizer – optimizer object