ascend-tools/train/atlas_benchmark-master/image_classification/DenseNet121/pytorch/README.md

DenseNet121_pytorch训练说明

1. 模型训练参数配置

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

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

---