Commits · ec2e8ffffc8b59f4fabf7b1f3f31861eb7a4af70 · OpenDAS / d2go

11 Mar, 2021 1 commit

update Lightning module test for OSS · 2b5a3176

Kai Zhang authored Mar 11, 2021

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/17

Use PyTorch Lightning checkpoint in the test.

Reviewed By: zhanghang1989

Differential Revision: D26962697

fbshipit-source-id: abe635e374c3ada130243f0eaadff34204f04fa1

2b5a3176

03 Mar, 2021 1 commit

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