Parcourir la source

修改vgg16训练脚本说明

liyan il y a 7 mois
Parent
commit
a90d79efeb
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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