他的回复:
回复:RoyalKun 发表于 2021-7-16 17:37 麻烦截图看下你的自定义镜像的训练作业的配置时怎样的?1.这是我在pycharm toolkit中的设置swr.cn-north-4.myhuaweicloud.com/ascend-share/5.0.2.alpha005_tensorflow-ascend910-cp37-euleros2.8-aarch64-training:1.15.0-2.0.12_0709/bin/bash run_train.sh 's3://bothdata/test/train/dense1_npu_20210711165332/' 'dense1_npu_20210711165332/boot_modelarts.py' '/tmp/log/train.log'2.这是创建相应文件夹和拷贝预训练模型的操作,即将OBS中的文件拷贝到 /home/work/.keras/models中。h5_path = os.path.join(config.modelarts_data_dir, "densenet169_weights_tf_dim_ordering_tf_kernels_notop.h5")cache_dir = os.path.join(os.path.expanduser('~'), '.keras/')datadir_base = os.path.expanduser(cache_dir)cache_subdir = 'models/'if not os.access(datadir_base, os.W_OK): datadir_base = os.path.join('/home/work/', '.keras/')datadir = os.path.join(datadir_base, cache_subdir)if not os.path.exists(datadir): os.makedirs(datadir)