Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "4a05bdfa869c80fdcac2d1b8fb48656f743a1fac"
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
lightning_train_net.py 8.31 KB