diffusion_models.utils.trainer

Classes

DiscriminativeTrainer(model, train_data, ...)
GenerativeTrainer(model, train_data, ...[, ...])
Trainer(model, train_data, loss_func, ...[, ...])	Trainer Class that trains 1 model instance on 1 device, suited for distributed training.