Преглед на файлове

修改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