tests/trainer/test_activation_checkpointing.py · 1c9e0e83b5b268cc7a90c5ac9d6142adb7d304da · OpenDAS / d2go

add warm up stage for d2go ema (for fsdp) · 1c9e0e83

Ji Hou authored Jul 25, 2023

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

per title

Reviewed By: wat3rBro

Differential Revision: D47740831

fbshipit-source-id: ecbe48a1085232a5cfb696e7f8e537d7e58e534a

1c9e0e83

test_activation_checkpointing.py 4.71 KB

Replace test_activation_checkpointing.py