Skip to content

Commit e396c7d

Browse files
authored
adapt for DCU (#3208)
1 parent ee09f87 commit e396c7d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tools/train.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,9 @@
2020
__dir__ = os.path.dirname(os.path.abspath(__file__))
2121
sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))
2222

23+
# XXX: avoid triggering error on DCU machines
24+
import tarfile
25+
2326
from ppcls.utils import config
2427
from ppcls.engine.engine import Engine
2528

0 commit comments

Comments
 (0)