1. 23 Apr, 2021 2 commits
  2. 22 Apr, 2021 1 commit
  3. 21 Apr, 2021 4 commits
  4. 20 Apr, 2021 2 commits
  5. 19 Apr, 2021 2 commits
    • Yue (R) Zhao's avatar
      tensorboard add_graph · 625cc2fa
      Yue (R) Zhao authored
      Summary: Add the API to log graph in tensorboard
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27855774
      
      fbshipit-source-id: 415c469c5de0c56fc828d1b95f4be697e0acac84
      625cc2fa
    • Peizhao Zhang's avatar
      Added hooks to report training progress to fblearner and keep alive. · bd6043ee
      Peizhao Zhang authored
      Summary:
      * Added a registry to register functions that could be used to register hooks for training.
        * TRAINER_HOOKS_REGISTRY: List of functions to add hooks for trainer, all functions in the registry will be called to add hooks
        * `func(hooks: List[HookBase]) -> None`
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D27560806
      
      fbshipit-source-id: fcfa02623bfd08508b6083db2d318d08f7e3c0b8
      bd6043ee
  6. 17 Apr, 2021 2 commits
    • Kai Zhang's avatar
      Delegate to model's customization · aeb24a92
      Kai Zhang authored
      Summary: Delegate FX quantization callback's customization to model.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27669212
      
      fbshipit-source-id: 2715546cf03134896da6f95ecddaf8503ff95d0b
      aeb24a92
    • Kai Zhang's avatar
      E2E QAT Workflow on Lightning · 845d0b2c
      Kai Zhang authored
      Summary:
      As per title and sanity test E2E QAT workflow on Lightning Trainer.
      
      - add `post_training_opts`. This is required to use `all_steps_qat.json` with Lightning. We don't actually support the post_training_opts in this diff though - we leave it part of T83437359.
      - Update .yaml to specify the Quantize-able modules.
      - Update `lightning_train_net.py` to use the QuantizationAwareTraining callback.
      
      Reviewed By: kandluis
      
      Differential Revision: D26304879
      
      fbshipit-source-id: 948bef4817d385d8a0969e4990d7f17ecd6994b7
      845d0b2c
  7. 15 Apr, 2021 3 commits
  8. 14 Apr, 2021 2 commits
  9. 13 Apr, 2021 2 commits
    • Yanghan Wang's avatar
      support rebuild train loader when data has expired · 754a95fa
      Yanghan Wang authored
      Summary:
      - store expiration in meta data when loading dat
      
      - use before_train_hook to rebuild data loader when expiration condition is met.
      
      Reviewed By: zisting
      
      Differential Revision: D27683164
      
      fbshipit-source-id: e3e3c6c15eee7c02c7a1bfed5f4d4d0e67d61a4f
      754a95fa
    • Sam Tsai's avatar
      update for supporting variation of coco-text like datasets · 2caa6646
      Sam Tsai authored
      Summary:
      1. Add changes to support variation of datasets
      2. Fix runner to support torchscript export
      
      Reviewed By: wat3rBro
      
      Differential Revision: D26871461
      
      fbshipit-source-id: ec46f7e0d2c14c9b802aec22d78b2a089e962a2f
      2caa6646
  10. 09 Apr, 2021 2 commits
  11. 08 Apr, 2021 1 commit
    • Yanghan Wang's avatar
      remove unused dataset_mapper.py · e47d6a24
      Yanghan Wang authored
      Summary: fbgs register_uri_image_loader, UniversalResourceLoader, _IMAGE_LOADER_REGISTRY returns no results other than this file
      
      Reviewed By: newstzpz
      
      Differential Revision: D27639902
      
      fbshipit-source-id: 52e3bb77dbb547334938b8537d6e1c173405d12d
      e47d6a24
  12. 06 Apr, 2021 2 commits
  13. 05 Apr, 2021 2 commits
    • Owen Wang's avatar
      make prediction count evaluation stable with DDP · cbdd46ab
      Owen Wang authored
      Summary: Prediction count evaluator needs to gather it's state before computing metrics, otherwise when parallelized across N GPUs, we only get metrics computed from 1/N of the dataset, increasing our eval signal's variance.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27416864
      
      fbshipit-source-id: b2c5334cd5a38bebcd06c6ace1627a6b71645fdd
      cbdd46ab
    • Sam Tsai's avatar
      add typing to transforms · 82f17be0
      Sam Tsai authored
      Summary: Add typing to transform.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27145140
      
      fbshipit-source-id: 8556427b421bf91a05692a590db175c68c4d6890
      82f17be0
  14. 03 Apr, 2021 2 commits
    • Peizhao Zhang's avatar
      Make data and evaluation visualization optional. · daf7f294
      Peizhao Zhang authored
      Summary:
      Make data and evaluation visualization optional.
      * could return None.
      
      Reviewed By: zhanghang1989, wat3rBro
      
      Differential Revision: D27316632
      
      fbshipit-source-id: 2a85db4815cbf3407a20a74c125dcd52d75167fa
      daf7f294
    • Peizhao Zhang's avatar
      Format changes. · 66df06ef
      Peizhao Zhang authored
      Summary:
      Format changes.
      * [Option] + [Shift] + [F]
      
      Reviewed By: mattcyu1, zhanghang1989, wat3rBro
      
      Differential Revision: D27316555
      
      fbshipit-source-id: 0fc3396eb34d964478cb3551dc73b47412089ccb
      66df06ef
  15. 02 Apr, 2021 1 commit
    • Yanghan Wang's avatar
      new interface for build_d2go_train_loader · 4bae056b
      Yanghan Wang authored
      Summary:
      #Facebook:
      
      `build_d2go_train_loader` will replace `runner.build_detection_train_loader`, currently we call `build_d2go_train_loader` from `runner.build_detection_train_loader` since some runner has there own implementation, we will solve those cases and remove the `runner.build_detection_train_loader` API.
      
      Currently `build_d2go_train_loader` uses `_MAPPED_TRAIN_LOADER_BUILDER_REGISTRY` to support different versions between OSS and FB, not sure if this is a good pattern or not, please comment in the diff if you have better idea.
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D27505681
      
      fbshipit-source-id: b5caf7280a88c2ebccb498097c0b7af51c966fc6
      4bae056b
  16. 31 Mar, 2021 3 commits
    • Kai Zhang's avatar
      Sync quantization callback changes · 1850a632
      Kai Zhang authored
      Reviewed By: newstzpz
      
      Differential Revision: D27255960
      
      fbshipit-source-id: 1699ff23d2bc610dffc0215a90a7c1c17e3783c3
      1850a632
    • Sam Tsai's avatar
      fix bypassed rebase error · 1027896a
      Sam Tsai authored
      Summary: Fixing unit test that was not listed due to rebase error.
      
      Reviewed By: newstzpz, wat3rBro
      
      Differential Revision: D27456322
      
      fbshipit-source-id: 519c5c086adfb19104ed99234f4f476eb34a79bc
      1027896a
    • Tao Xu's avatar
      Train a pix2pix model · bb0ea3d5
      Tao Xu authored
      Summary: Train a pix2pix model on the paired dataset. During inference, it can transfer an source image to the target image.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27371290
      
      fbshipit-source-id: 3141bc6d9e4fe0013f6ea3de3cf998163d286168
      bb0ea3d5
  17. 30 Mar, 2021 3 commits
  18. 29 Mar, 2021 4 commits