DL-Art-School/codes/models
James Betker 3074f41877 Get rosinality model converter to work
Mostly, just needed to remove the custom cuda ops, not so bueno on Windows.
2020-12-17 16:03:39 -07:00
..
archs Get rosinality model converter to work 2020-12-17 16:03:39 -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 Rosinality stylegan2 port 2020-12-17 14:18:46 -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 Rosinality stylegan2 port 2020-12-17 14:18:46 -07:00