James Betker ea56eb61f0 Fix DDP errors for discriminator ... - Don't define training_net in define_optimizers - this drops the shell and leads to problems downstream - Get rid of support for multiple training nets per opt. This was half baked and needs a better solution if needed downstream.		2020-12-07 12:50:57 -07:00
..
archs	Revert gaussian loss changes	2020-12-04 12:49:20 -07:00
eval	Go back to apex DDP, fix distributed bugs	2020-12-04 16:39:21 -07:00
experiments	Add in experiments hook	2020-09-19 10:05:25 -06:00
flownet2@db2b7899ea	Update flownet submodule	2020-10-24 11:59:00 -06:00
steps	Fix DDP errors for discriminator	2020-12-07 12:50:57 -07:00
__init__.py	Lots of new discriminator nets	2020-11-10 16:06:54 -07:00
base_model.py	Go back to apex DDP, fix distributed bugs	2020-12-04 16:39:21 -07:00
ExtensibleTrainer.py	Go back to apex DDP, fix distributed bugs	2020-12-04 16:39:21 -07:00
feature_model.py	Feature mode -> back to LR fea	2020-09-11 13:09:55 -06:00
loss.py	More adjustments to support multi-modal training	2020-10-22 16:49:34 -06:00
lr_scheduler.py	Extensible trainer (in progress)	2020-08-12 08:45:23 -06:00
networks.py	Cleanup	2020-12-03 23:42:51 -07:00