AI开发平台MODELARTS-使用自定义数据集进行训练:Step2 修改配置文件

时间:2024-05-28 16:22:36

Step2 修改配置文件

配置文件主要包括算法配置文件和算法参数文件,例如:

  • 算法配置文件:./algorithms/ivgPose/config/sample/config.py
  • 算法参数文件:./algorithms/ivgPose/config/sample/simplepose_resnet50_coco_256x192.py

根据需要修改训练相关参数,如数据集路径data_root、runner中的train_args参数以及train参数等。如下述代码所示。

config.py文件

   
    train_args=dict(
        cfg=alg_cfg['cfg'],
        output_dir=run_dir,
        data_root=alg_cfg['data_root'],
        load_from=None,
        gpus=None,
        lr=None,
        max_epoch=5,
        samples_per_gpu=None,
        resume_from=None,
        pretrained=alg_cfg['pretrained'],
        train_data_root=None,
        val_data_root=None,
    ),

simplepose_resnet50_coco_256x192.py文件

    
train = dict(
    save_period=-1,
    val_period=5,
    best_saver=dict(key_indicator='AP', rule='max'),
    ema=dict(type='ExpEMA')
)
support.huaweicloud.com/devtool-modelarts/devtool-modelarts_0178.html