1. 30 Mar, 2021 1 commit
    • Sam Tsai's avatar
      reorganize unit tests · a0658c4a
      Sam Tsai authored
      Summary: Separate unit tests into individual folder based on functionality.
      
      Reviewed By: wat3rBro
      
      Differential Revision: D27132567
      
      fbshipit-source-id: 9a8200be530ca14c7ef42191d59795b05b9800cc
      a0658c4a
  2. 24 Mar, 2021 2 commits
    • Kai Zhang's avatar
      Support evaluate predictor · 6aec097e
      Kai Zhang authored
      Summary:
      Evaluate the predictor generated by previous step.
      This diff modify the lightning_train_net to reuse the evaluation logic by adding a `predictor_path` param.
      This diff also makes Lightning training backend depends on `cfg.MODEL.DEVICE` so that in evaluate_predictor step, user could set backend by changing model device. This is useful for evaluating int8 quantized model.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27150609
      
      fbshipit-source-id: fb72da3e81db932c0fa479350150720143e09a3e
      6aec097e
    • Kai Zhang's avatar
      Simplify Lightning task and model creation · 9051f71a
      Kai Zhang authored
      Summary:
      Given that the way to create D2 (https://github.com/facebookresearch/d2go/commit/465cdb842513eb910aa20fcedea1d2edd15dc7b7)go runner and Lightning task are different, get_class was introduced so that in application we could do:
      ```
      if is Lightning:
          task_cls = get_class(classname)
          task = task_cls(cfg)
      else:
          runner = create_runner(classname)
      ```
      It turns out that we could need to do that in many places: workflow, binaries.
      This diff revert `get_class` and return class in `create_runner` if the class is a Lightning module.
      
      Reviewed By: newstzpz
      
      Differential Revision: D26676595
      
      fbshipit-source-id: c3ce2016d09fe073af4c2dd9f98eea4e59ca621b
      9051f71a
  3. 20 Mar, 2021 1 commit
    • Yanghan Wang's avatar
      move test utils to core library · 9d238344
      Yanghan Wang authored
      Summary: Not d2go.tests is not a library for oss, move utils code to d2go.utils.testing
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D26706933
      
      fbshipit-source-id: 85767b66bbb6c67db05e11823beb4840220b2aa3
      9d238344
  4. 18 Mar, 2021 1 commit
  5. 16 Mar, 2021 1 commit
    • Sam Tsai's avatar
      extend cocotext conversion · b80f2ff5
      Sam Tsai authored
      Summary: Extend conversion to support ids beyond cocotext format where ids are strings.
      
      Reviewed By: newstzpz
      
      Differential Revision: D27018211
      
      fbshipit-source-id: 7282fd4b9a7e9cd19323235ed1a3c3e7b33cb6b4
      b80f2ff5
  6. 11 Mar, 2021 1 commit
  7. 05 Mar, 2021 1 commit
    • Peizhao Zhang's avatar
      migrated to iopath for d2go - rest. · 614fdd32
      Peizhao Zhang authored
      Summary: migrated to iopath for d2go - rest.
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D26632160
      
      fbshipit-source-id: 182fcd9ad493a483ab7f342f373a7be2de7d10ca
      614fdd32
  8. 04 Mar, 2021 1 commit
  9. 03 Mar, 2021 3 commits
    • Kai Zhang's avatar
      Split lightning_train_net into OSS and internal · 857195d8
      Kai Zhang authored
      Summary:
      As titled. The OSS version only use PyTorch Lightning while internal version leverages some features(e.g. Manifold integration, every_n_step checkpointing).
      This diff splits train_net.main into smaller functions so that they could be shared across OSS and internal versions.
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D26752701
      
      fbshipit-source-id: 7f68e2a81e78193e117517a0ff668ab14b76ea65
      857195d8
    • Kai Zhang's avatar
      Copy quantization callback to D2go · 5d8068d8
      Kai Zhang authored
      Summary: As titled. Make a copy of quantization callback to unblock D2go OSS.
      
      Reviewed By: zhanghang1989
      
      Differential Revision: D26735525
      
      fbshipit-source-id: 12b77f04cfa1361e856b26ea218a262da1fadd88
      5d8068d8
    • facebook-github-bot's avatar
      Initial commit · f23248c0
      facebook-github-bot authored
      fbshipit-source-id: f4a8ba78691d8cf46e003ef0bd2e95f170932778
      f23248c0