Skip to content

Commit b77f426

Browse files
committed
adapt for DCU (#3208)
1 parent 6256dd7 commit b77f426

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)