VGG用PyTorch实现后在控制台显示模型参数?
时间: 2024-05-14 18:13:27 浏览: 141
VGG网络--基于pytorch实现
你可以使用以下代码来在控制台显示VGG模型的参数:
```python
import torch
from torchvision import models
vgg = models.vgg16(pretrained=False)
print(vgg.parameters())
```
这将输出VGG模型的所有参数。
阅读全文
VGG网络--基于pytorch实现