diff --git a/tools/train.py b/tools/train.py index e7c9d7bcc..2214ab320 100644 --- 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