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

integrated tensorboardx test-tube

Browse Source
This commit is contained in:
William Falcon
2019-06-29 15:58:47 -04:00
parent 38c9102d13
commit f2134a4ddd
3 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/root_module.py
+1
View File
@@ -26,6 +26,7 @@ class LightningModule(GradInformation, ModelIO, OptimizerConfig, ModelHooks):
self.gradient_clip = hparams.gradient_clip
self.trainer = None
self.from_lightning = True
self.experiment = None
# track if gpu was requested for checkpointing
self.on_gpu = False