4. torchgan.trainer

This subpackage provides ability to perform end to end training capabilities of the Generator and Discriminator models. It provides strong visualization capabilities using tensorboardX. In most cases you will need to overwrite the generator_train_iter() and discriminator_train_iter().

Currently supported Trainers include: