Add FSDP support to MNISTLitModule
(#604)
#539
Job | Run time |
---|---|
5m 52s | |
4m 11s | |
4m 36s | |
4m 17s | |
4m 27s | |
4m 3s | |
4m 42s | |
8m 12s | |
7m 37s | |
4m 0s | |
51m 57s |
MNISTLitModule
(#604)
#539
Job | Run time |
---|---|
5m 52s | |
4m 11s | |
4m 36s | |
4m 17s | |
4m 27s | |
4m 3s | |
4m 42s | |
8m 12s | |
7m 37s | |
4m 0s | |
51m 57s |