mmdetection传到OBS中
我们需要在训练作业中安装mmdetection。有两种方法,一种是存在OBS中,然后传入训练作业中,另外一种使用代码从github上下载。这里,我们使用第一种方法,因为这样子,我门还可以对mmdetection进行修改。所以,我们先下载mmdetection,然后上传到OBS中。并将此目录作为代码目录。
写启动代码
mmdetection内包括train.py和test.py的启动代码。这里我们不选择修改这两个py,而是新写一个main.py,作为启动代码,将其放在mmdetection的目录下。
import moxing as mox
import os
# 接受训练作业传入的两个参数
import argparse
parser = argparse.ArgumentParser(description='training process')
parser.add_argument('--train_url', type=str, default='') # 训练作业的输出位置
parser.add_argument('--data_url', type=str, default='') # data在obs上的地址
args, _ = parser.parse_known_args(