wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-12 12:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed correct module on hpc save

Browse Source
This commit is contained in:
William Falcon
2019-07-24 18:16:02 -04:00
parent 7217ecdb18
commit d7be0aae1c
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/README.md
+7
View File
@@ -36,4 +36,11 @@ This file fits a tiny model on MNIST using these different set-ups.
3. Multiple (2) GPUs using DP + apex (for 16-bit precision).
3. Multiple (2) GPUs using DDP + apex (for 16-bit precision).
For each set up it also tests:
1. model saving.
2. model loading.
3. predicting with a loaded model.
4. simulated save from HPC signal.
5. simulated load from HPC signal.