Add simple FSDP support to MNIST example LightningModule
#538
Job | Run time |
---|---|
4m 49s | |
4m 48s | |
4m 7s | |
5m 9s | |
4m 10s | |
4m 31s | |
4m 10s | |
4m 33s | |
4m 20s | |
4m 16s | |
44m 53s |
LightningModule
#538
Job | Run time |
---|---|
4m 49s | |
4m 48s | |
4m 7s | |
5m 9s | |
4m 10s | |
4m 31s | |
4m 10s | |
4m 33s | |
4m 20s | |
4m 16s | |
44m 53s |