DL-Art-School/codes/models
James Betker 0a19e53df0 BYOL mods
2020-12-14 23:59:11 -07:00
..
archs Allow RRDB to upscale 8x 2020-12-14 23:58:52 -07:00
byol BYOL mods 2020-12-14 23:59:11 -07:00
eval Directly use dataset keys 2020-12-04 20:14:53 -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 Directly use dataset keys 2020-12-04 20:14:53 -07:00
feature_model.py Directly use dataset keys 2020-12-04 20:14:53 -07: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 BYOL with structure! 2020-12-10 15:07:35 -07:00