1. 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
  2. 17 Mar, 2021 1 commit
  3. 09 Mar, 2021 1 commit
    • Yanghan Wang's avatar
      add benchmark_data binary · 66b7c7c8
      Yanghan Wang authored
      Reviewed By: newstzpz
      
      Differential Revision: D26072333
      
      fbshipit-source-id: 6727b34458d410e904045aa58f81c3e09111882a
      66b7c7c8
  4. 07 Mar, 2021 1 commit
  5. 04 Mar, 2021 1 commit
    • RangiLyu's avatar
      Typo fixes · 5bf4cc7d
      RangiLyu authored
      Summary:
      Change depoyment to deployment in README.md.
      Change datasest to datasets in tools/exporter.py.
      
      Pull Request resolved: https://github.com/facebookresearch/d2go/pull/7
      
      Reviewed By: newstzpz
      
      Differential Revision: D26821039
      
      Pulled By: zhanghang1989
      
      fbshipit-source-id: 5056d15c877c4b3d771d33267139e73f1527da21
      5bf4cc7d
  6. 03 Mar, 2021 2 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
    • facebook-github-bot's avatar
      Initial commit · f23248c0
      facebook-github-bot authored
      fbshipit-source-id: f4a8ba78691d8cf46e003ef0bd2e95f170932778
      f23248c0