ertk.pytorch.train

Training configuration for PyTorch models.

PyTorchTrainConfig([batch_size, logging, ...])
PyTorchLoggingConfig([log_dir, tensorboard, csv])
PyTorchDataConfig(train_data_processing, ...)
PyTorchDataAugConfig(spec, kwargs, typing.Any] =)