mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
easy import for lightningModule
This commit is contained in:
1 parent
7d08e52b5d
commit
1e57a75ff9
1 file changed
+1
-1
@@ -346,7 +346,7 @@ class Trainer(TrainerIO):
|
||||
for i in range(0, 120):
|
||||
sleep(1.0)
|
||||
if os.path.exists(ip_file):
|
||||
ip = open(file=ip_file, mode='r')
|
||||
ip = list(open(file=ip_file, mode='r'))[0]
|
||||
return ip
|
||||
|
||||
def __run_pretrain_routine(self, model):
|
||||
|
||||
Reference in new issue
Block a user