DL-Art-School/codes/models
2020-12-16 10:27:38 -07:00
..
archs Clean up of SRFlowNet_arch 2020-12-16 10:27:38 -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 Spinenet with logits head 2020-12-15 17:16:19 -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 Allow swapping to torch DDP as needed in code 2020-12-09 15:03:59 -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 Spinenet with logits head 2020-12-15 17:16:19 -07:00