更新 'train_for_c2net.py'

2022-06-23 15:34:17 +08:00 · 2022-06-23 15:34:17 +08:00 · 1c75273032
parent 5b1076719d
commit 1c75273032
1 changed files with 2 additions and 2 deletions
--- 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,