提交 cbc970fc 编写于 作者: Y Yang Zhang 提交者: GitHub

Make weight download hack compatible with python 2 (#3432)

上级 f83561e4
...@@ -17,6 +17,7 @@ from __future__ import division ...@@ -17,6 +17,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
from __future__ import unicode_literals from __future__ import unicode_literals
import errno
import os import os
import shutil import shutil
import time import time
...@@ -57,9 +58,13 @@ def _get_weight_path(path): ...@@ -57,9 +58,13 @@ def _get_weight_path(path):
weight_path = map_path(path, WEIGHTS_HOME) weight_path = map_path(path, WEIGHTS_HOME)
lock_path = weight_path + '.lock' lock_path = weight_path + '.lock'
if not os.path.exists(weight_path): if not os.path.exists(weight_path):
os.makedirs(os.path.dirname(weight_path), exist_ok=True) try:
os.makedirs(os.path.dirname(weight_path))
except OSError as e:
if e.errno != errno.EEXIST:
raise
with open(lock_path, 'w'): # touch with open(lock_path, 'w'): # touch
os.utime(lock_path) os.utime(lock_path, None)
if trainer_id == 0: if trainer_id == 0:
get_weights_path(path) get_weights_path(path)
os.remove(lock_path) os.remove(lock_path)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册