adapt for DCU (#3208)

2024-08-07 14:26:50 +08:00 · 2024-08-07 14:26:50 +08:00 · e396c7d30b
parent ee09f87d9b
commit e396c7d30b
1 changed files with 3 additions and 0 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -20,6 +20,9 @@ import sys
 __dir__ = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(os.path.abspath(os.path.join(__dir__, '../')))

+# XXX: avoid triggering error on DCU machines
+import tarfile
+
 from ppcls.utils import config
 from ppcls.engine.engine import Engine