- 31 Jul, 2019 1 commit
-
-
Francisco Massa authored
* Copy classification scripts for video classification * Initial version of video classification * add version * Training of r2plus1d_18 on kinetics work Gives even slightly better results than expected, with 57.336 top1 clip accuracy. But we count some clips twice in this evaluation * Cleanups on training script * Lint * Minor improvements * Remove some hacks * Lint
-
- 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
-