ascend-tools/train/atlas_benchmark-master/image_classification/DeepMar/pytorch

DeepMar_pytorch训练说明

1. 数据集处理

1.1. 下载并准备数据集：

百度云盘https://pan.baidu.com/s/1q8nsydT7xkDjZJOxvPcoEw passwd: 5vep 或者https://drive.google.com/open?id=1q4cux17K3zNBgIrDV4FtcHJPLzXNKfYG

存放地址 ./dataset/peta/images/*.png ./dataset/peta/PETA.mat

1.2 运行以下命令，分割训练集、测试集（路径修改成自己存放数据集路径）

python script/dataset/transform_peta.py 生成 peta_dataset.pkl，peta_partition.pkl 文件

2. 模型训练参数配置

在train/yaml/DeepMar.yaml中修改相应配置， 配置项含义:

pytorch_config:
    data_url: 数据集路径
    epoches: 跑多少个epoch
    batch_size:1p 参数为256 2p 512 4p 1024  8p为2048
    seed: 49
    lr: 默认参数1p 0.01 2p 0.016 4p 0.016 8p 0.016
    docker_image: docker 镜像名称:版本号

---