Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • 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
test_lightning_train_net.py 1.87 KB