@@ -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