GitLab

* [WIP] Update ref example video classification

* [WIP] Updated video classification ref example

* Replaced mem format conversion functions by classes

* remove sys.version_info == 2

* remove sys.version_info < 3

* remove from __future__ imports

* Generalised for custom dataset

* Typo, redundant code, sensible default

* Args for name of train and val dir

* move sampler into TV core. Update UniformClipSampler

* Fix reference training script

* Skip test if pyav not available

* change interpolation from round() to floor() as round(0.5) behaves differently between py2 and py3

* [WIP] Minor cleanups on R3d

* Move all models to video/resnet.py

* Remove old files

* Make tests less memory intensive

* Lint

* Fix typo and add pretraing arg to training script

* Move RandomClipSampler to references

* Lint and bugfix

Also add docs

* 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

* adding mixed precision training with Apex

* fix APEX default optimization level

* adding python version check for apex

* fix LINT errors and raise exceptions if apex not available

* fixing apex distributed training

* fix throughput calculation: include forward pass

* remove torch.cuda.set_device(args.gpu) as it's already called in init_distributed_mode

* fix linter: new line

* move Apex initialization code back to the beginning of main

* move apex initialization to before lr_scheduler - for peace of mind. Though, doing apex initialization after lr_scheduler seems to work fine as well

* adding mixed precision training with Apex

* fix APEX default optimization level

* adding python version check for apex

* fix LINT errors and raise exceptions if apex not available

Allows for easily evaluating the pre-trained models in the modelzoo

* Miscellaneous improvements to the classification reference scritps

* Fix lint

* Add groups support to ResNet

* Kill BaseResNet

* Make it support multi-machine training

Making references/classification/train.py and references/classification/utils.py  compatible with python2  (#831)

* linter fixes

* linter fixes

* 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