From 1c75273032c8af311e40d035b23448c19ed8b56b Mon Sep 17 00:00:00 2001 From: lewis <747342561@qq.com> Date: Thu, 23 Jun 2022 15:34:17 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'train=5Ffor=5Fc2net.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train_for_c2net.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_for_c2net.py b/train_for_c2net.py index 728d9ec..8356a9f 100644 --- a/train_for_c2net.py +++ b/train_for_c2net.py @@ -38,8 +38,8 @@ if __name__ == "__main__": print('args:') print(args) - train_dir = '/tmp/output' - data_dir = '/tmp/dataset' + train_dir = '/cache/output' + data_dir = '/cache/dataset' #注意:这里很重要,指定了训练所用的设备CPU还是Ascend NPU context.set_context(mode=context.GRAPH_MODE,