tools/train_net.py · 7ad54f578f93dc36407c2907f5d2e15a81a55f01 · OpenDAS / d2go

Add Optimizer FSDP and AC on 3xUnet/5xUnet · 7ad54f57

Jessica Zhong authored Aug 24, 2023

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

Reviewed By: wat3rBro, YanjunChen329

Differential Revision: D48544742

fbshipit-source-id: 9e49f13aa50e065c30e5551a636a83afd2d11acd

7ad54f57

train_net.py 6.42 KB

Replace train_net.py