easy import for lightningModule

2026-09-10 12:21:57 +08:00 · 2019-07-08 09:32:57 -04:00
parent 7d08e52b5d
commit 1e57a75ff9
1 changed files with 1 additions and 1 deletions
@@ -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):