From 8facc642f2367497138eef7501e8f835d2ea28ba Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 26 Jun 2019 19:50:53 -0400 Subject: [PATCH] added docs page --- README.md | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index cfca75bf7d..d12da08ec3 100644 --- a/README.md +++ b/README.md @@ -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).