Commits · dbc9a9f4b635b9c51de1dd405dbca4fdff78614e · OpenDAS / d2go

03 Mar, 2021 4 commits

Enable manifold based training for OCR d2go pipeline · dbc9a9f4

Rama Kovvuri authored Mar 03, 2021

Summary: As in title.

Reviewed By: SuperIRabbit

Differential Revision: D25349762

fbshipit-source-id: 69e21cc00a9dd5b70a479fd4383742593a62997e

dbc9a9f4

Split lightning_train_net into OSS and internal · 857195d8

Kai Zhang authored Mar 03, 2021

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

Copy quantization callback to D2go · 5d8068d8

Kai Zhang authored Mar 03, 2021

Summary: As titled. Make a copy of quantization callback to unblock D2go OSS.

Reviewed By: zhanghang1989

Differential Revision: D26735525

fbshipit-source-id: 12b77f04cfa1361e856b26ea218a262da1fadd88

5d8068d8

Initial commit · f23248c0
facebook-github-bot authored Mar 02, 2021
```
fbshipit-source-id: f4a8ba78691d8cf46e003ef0bd2e95f170932778
```
f23248c0