1. 03 Mar, 2021 5 commits
    • Yuxin Wu's avatar
      clean up reference of FBTensorboardXWriter · 465cdb84
      Yuxin Wu authored
      Reviewed By: theschnitz
      
      Differential Revision: D26230057
      
      fbshipit-source-id: 57eb15b5eea27f65035b2509e920ba269f7999f1
      465cdb84
    • Rama Kovvuri's avatar
      Enable manifold based training for OCR d2go pipeline · dbc9a9f4
      Rama Kovvuri authored
      Summary: As in title.
      
      Reviewed By: SuperIRabbit
      
      Differential Revision: D25349762
      
      fbshipit-source-id: 69e21cc00a9dd5b70a479fd4383742593a62997e
      dbc9a9f4
    • 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