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,