added module properties

This commit is contained in:
William Falcon 2019-06-29 18:14:45 -04:00
parent 04445504e5
commit 7fd2b0fa19
1 changed files with 18 additions and 3 deletions

View File

@ -8,6 +8,15 @@ The current epoch
#### dtype #### dtype
Current dtype Current dtype
---
#### experiment
An instance of test-tube Experiment which you can use to log anything for tensorboarX.
```{.python}
self.experiment.add_embedding(...)
self.experiment.log({'val_loss': 0.9})
self.experiment.add_scalars(...)
```
--- ---
#### global_step #### global_step
Total training batches seen across all epochs Total training batches seen across all epochs
@ -21,5 +30,11 @@ The current gradient clip value
True if your model is currently running on GPUs. Useful to set flags around the LightningModule for different CPU vs GPU behavior. True if your model is currently running on GPUs. Useful to set flags around the LightningModule for different CPU vs GPU behavior.
--- ---
#### Trainer #### trainer
Last resort access to any state the trainer has. Changing certain properties here could affect your training run. Last resort access to any state the trainer has. Changing certain properties here could affect your training run.
```{.python}
self.trainer.optimizers
self.trainer.current_epoch
...
```