forked from ecker/DL-Art-School
036684893e
- Added LARS and SGD optimizer variants that support turning off certain features for BN and bias layers - Added a variant of pytorch's resnet model that supports gradient checkpointing. - Modify the trainer infrastructure to support above - Fix bug with BYOL (should have been nonfunctional) |
||
|---|---|---|
| .. | ||
| byol | ||
| fixup_resnet | ||
| flownet2@db2b7899ea | ||
| glean | ||
| srflow | ||
| stylegan | ||
| switched_conv@cb520afd4d | ||
| tecogan | ||
| transformers | ||
| __init__.py | ||
| arch_util.py | ||
| discriminator_vgg_arch.py | ||
| feature_arch.py | ||
| ProgressiveSrg_arch.py | ||
| ResGen_arch.py | ||
| resnet_with_checkpointing.py | ||
| RRDBNet_arch.py | ||
| spinenet_arch.py | ||
| srg2_classic.py | ||
| SwitchedResidualGenerator_arch.py | ||