浏览代码

修改vgg16训练脚本说明

liyan 7 月之前
父节点
当前提交
a90d79efeb
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      README.md

+ 1 - 1
README.md

@@ -42,7 +42,7 @@ python train_alexnet.py --data-path dataset/imagenette2-320 --output-dir checkpo
 ```
 - VGG16
 ```shell
-python train_vgg16.py --data-path dataset/imagenette2-320 --output-dir checkpoints/vgg16 --batch-size 64 --epochs 90
+python train_vgg16.py --data-path dataset/imagenette2-320 --output-dir checkpoints/vgg16 --batch-size 64 --epochs 90 --lr 0.01 --opt sgd
 ```
 
 ## 模型训练权重转换为onnx