Sfoglia il codice sorgente

修改vgg16训练脚本说明

liyan 7 mesi fa
parent
commit
a90d79efeb
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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