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

added docs page

Browse Source
This commit is contained in:
William Falcon
2019-06-26 19:50:53 -04:00
parent 71af821b85
commit 8facc642f2
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+7 -5
View File
@@ -107,13 +107,15 @@ def load_model_specific(self, checkpoint):
self.load_state_dict(checkpoint['state_dict'])
```
- Log all details of your experiment (model params, code snapshot, etc...)
```python
from test_tube import Experiment
exp = Experiment(...)
Trainer(experiment=exp)
```
5. Learning rate annealing.
6. Can train complex models like GANs or anything with multiple optimizers.
7. Weight checkpointing.
8. Model saving.
9. Model loading.
10. Log training details (through test-tube).
11. Run training on multiple GPUs (through test-tube).
12. Run training on a GPU cluster managed by SLURM (through test-tube).