diff --git a/train.py b/train.py index 50bc69d..d338a86 100644 --- a/train.py +++ b/train.py @@ -10,8 +10,8 @@ train lenet and get network model files(.ckpt) args.data_url = '/home/ma-user/work/data/' //数据集位置 args.train_url = '/home/ma-user/work/model/' //训练输出的模型位置 在训练环境变换为: -args.data_url = '/home/work/user-job-dir/inputs/data/' -args.train_url = '/home/work/user-job-dir/outputs/model/' +args.data_url = '/home/work/user-job-dir/data/' +args.train_url = '/home/work/user-job-dir/model/' 2、在训练环境中,需要将数据集从obs拷贝到训练镜像中,训练完以后,需要将输出的模型拷贝到obs. 将数据集从obs拷贝到训练镜像中: