关键字:昇腾 头孢 云训练
Baseline
传送门:Github上mindspore 0.5版的YOLOv3
适配云端训练
Baseline的代码在本地训练的话,不需要更改。但在ModelArts上训练的话,需要适配一下。
首先,需要引用一个库方便OBS数据Copy。还要声明两个地址
import moxing as mox
local_data_url = '/cache/data'
local_train_url = '/cache/train_url'
其次,需要在parser增加两个参数,如下。
parser.add_argument("--data_url"