GitLab

Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Add is_qat support using a method getter

* Switch to an internal _fuse_modules

* Fix linter.

* Pass is_qat=False on PTQ

* Fix bug on ra_sampler flag.

* Set is_qat=True for QAT

Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* [references] Adding gradient clipping

* ufmt formatting

* remove apex code

* resolve naming issue

Co-authored-by: Nicolas Hug <nicolashug@fb.com>

* Update EMA every X iters.

* Adding AdamW optimizer.

* Adjusting EMA decay scheme.

* Support custom weight decay for Normalization layers.

* Fix identation bug.

* Change EMA adjustment.

* Quality of life changes to faciliate testing

* ufmt format

* Fixing imports.

* Adding FixRes improvement.

* Support EMA in store_model_weights.

* Adding interpolation values.

* Change train_crop_size.

* Add interpolation option.

* Removing hardcoded interpolation and sizes from the scripts.

* Fixing linter.

* Incorporating feedback from code review.

* add ufmt as code formatter

* cleanup

* quote ufmt requirement

* split imports into more groups

* regenerate circleci config

* fix CI

* clarify local testing utils section

* use ufmt pre-commit hook

* split relative imports into local category

* Revert "split relative imports into local category"

This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51.

* pin black and usort dependencies

* fix local test utils detection

* fix ufmt rev

* add reference utils to local category

* fix usort config

* remove custom categories sorting

* Run pre-commit without fixing flake8

* got a double import in merge
Co-authored-by: Nicolas Hug <nicolashug@fb.com>

* Add RandomMixupCutmix.

* Add test with real data.

* Use dataloader and collate in the test.

* Making RandomMixupCutmix JIT scriptable.

* Move out label_smoothing and try roll instead of flip

* Adding mixup/cutmix in references script.

* Handle one-hot encoded target in accuracy.

* Add support of devices on tests.

* Separate Mixup from Cutmix.

* Add check for floats.

* Adding device on expect value.

* Remove hardcoded weights.

* One-hot only when necessary.

* Fix linter.

* Moving mixup and cutmix to references.

* Final code clean up.

* Added Exponential Moving Average support to classification reference script

* Addressed review comments

* Updated model argument

* Adding the average_checkpoints() method.

* Adding the store_model_weights() method.

* remove sys.version_info == 2

* remove sys.version_info < 3

* remove from __future__ imports

* 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()'

* 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