- 10 Feb, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Adding the average_checkpoints() method. * Adding the store_model_weights() method.
-
- 31 Mar, 2020 1 commit
-
-
Philip Meier authored
* remove sys.version_info == 2 * remove sys.version_info < 3 * remove from __future__ imports
-
- 14 Jun, 2019 1 commit
-
-
LXYTSOS authored
* can't work with pytorch-cpu fixed utils.py can't work with pytorch-cpu because of this line of code `memory=torch.cuda.max_memory_allocated()` * can't work with pytorch-cpu fixed utils.py can't work with pytorch-cpu because of this line of code 'memory=torch.cuda.max_memory_allocated()'
-
- 08 May, 2019 1 commit
-
-
Francisco Massa authored
* Miscellaneous improvements to the classification reference scritps * Fix lint
-
- 02 Apr, 2019 2 commits
-
-
Francisco Massa authored
* Add groups support to ResNet * Kill BaseResNet * Make it support multi-machine training
-
Surgan Jandial authored
Making references/classification/train.py and references/classification/utils.py compatible with python2 (#831) * linter fixes * linter fixes
-
- 28 Mar, 2019 1 commit
-
-
Francisco Massa authored
* Initial version of classification reference training script * Updates * Minor updates * Expose a few more options * Load optimizer and lr_scheduler when resuming Also log the learning rate * Evaluation-only and minor improvements Identified a bug in the reporting of the results. They need to be reduced between all processes * Address Soumith's comment * Fix some approximations on the evaluation metric * Flake8
-