Revision 0ee1127c81bf245225cd7db50d631e8677abaefc authored by Sasha Sheng on 08 February 2021, 10:50:33 UTC, committed by Facebook GitHub Bot on 08 February 2021, 10:52:20 UTC
Summary:
* pytorch lighting stub mostly involving training
* Tests for lightning trainer included
* built on top of the mmf grad accumulation fix: https://github.com/facebookresearch/mmf/pull/747

- [X] MVP 0. Training: Goal - Train a model from scratch and reach similar accuracy as using mmf_trainer
   - [X] Setup the training pipeline: done
   - [X] Training on the right device: done
   - [X] Clip gradients: done
   - [X] Optimizer: done
   - [X] FP16 Support: done
   - [X] LR scheduler (incl. warmup etc): done
   - [X] testcase: train visual_bert on vqa from scratch fo 10 iterations, compare the value: done
- [x] tests included in this PR (tests are only done for pytorch lightning integration):
   - [X] Vanilla Training w/o grad accumulate, make sure loss for 5 iters are the same between mmf and pl
      - [X] Optimizer working as intended as a part of this PR.
   - [X] `max_updates` and `max_epochs` calculation
   - [x] Training with grad accumulate
   - [x] Training with LR schedule achieves a different value compared to without LR schedule
   - [x] Training with LR schedule for PL is the same as training with LR schedule for `mmf_tranier`
   - [x] Training with gradient clipping make sure all grads are within the `grad_clipping` threshold
   - [x] Training with gradient clipping is the same as training with gradient clipping for `mmf_trainer`

Pull Request resolved: https://github.com/facebookresearch/mmf/pull/748

Reviewed By: apsdehal, simran2905

Differential Revision: D26192869

Pulled By: ytsheng

fbshipit-source-id: 203a91e893d6b878bbed80ed84960dd059cfc90c
1 parent fc72ef0
History
File Mode Size
docs
src
static
.eslintignore -rw-r--r-- 30 bytes
.eslintrc.js -rw-r--r-- 1.6 KB
.gitignore -rw-r--r-- 282 bytes
.prettierignore -rw-r--r-- 42 bytes
.prettierrc -rw-r--r-- 180 bytes
.stylelintrc.js -rw-r--r-- 305 bytes
README.md -rw-r--r-- 998 bytes
build_docs.sh -rwxr-xr-x 2.2 KB
docusaurus.config.js -rw-r--r-- 3.0 KB
package.json -rw-r--r-- 1.4 KB
sidebars.js -rw-r--r-- 1.2 KB
yarn.lock -rw-r--r-- 495.4 KB

README.md

back to top