easy import for lightningModule

This commit is contained in:
William Falcon
2019-07-08 09:30:51 -04:00
parent d540d476a0
commit 7d08e52b5d
-2
View File
@@ -332,8 +332,6 @@ class Trainer(TrainerIO):
print(ip_file_dir)
ip_file = os.path.join(ip_file_dir, '.ip_meta')
if proc_rank == 0:
os.makedirs(ip_file, exist_ok=True)
# get the proc 0 IP
root_ip = subprocess.run(['hostname', '-I'], stdout=subprocess.PIPE).stdout.decode('utf-8')
root_ip = root_ip.split(' ')[0]