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

pointer to trainer in model

Browse Source
This commit is contained in:
William Falcon
2019-04-23 07:25:09 -04:00
parent b625b293f4
commit 676d76d839
3 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/root_module.py
+1
View File
@@ -24,6 +24,7 @@ class RootModule(GradInformation, ModelIO, OptimizerConfig, ModelHooks):
self.overfit = hparams.overfit
self.gradient_clip = hparams.gradient_clip
self.num = 2
self.trainer = None
# track if gpu was requested for checkpointing
self.on_gpu = False