EfficientNet_V2_S_Weights预训练权重

### 下载 PyTorch 中 EfficientNet V2 S 预训练权重

为了下载并使用 `EfficientNet_V2_S` 模型的预训练权重，在最新版本的 `torchvision` 库中可以直接通过模型构建函数获取这些权重。下面展示了如何实现这一过程：

import torchvision.models as models

# 加载带有 ImageNet 上预训练权重的 EfficientNet V2 Small 模型
model = models.efficientnet_v2_s(weights='IMAGENET1K_V1')

如果希望仅下载权重而不立即加载整个模型结构，则可以采用如下方式保存权重到本地文件系统以便后续使用[^2]:

from torchvision import models
import torch

weights = models.EfficientNet_V2_S_Weights.IMAGENET1K_V1
state_dict = weights.get_state_dict(progress=True)

# 将权重字典保存至 .pth 文件
torch.save(state_dict, 'efficientnet_v2s_imagenet1k_weights.pth')

当需要重新加载已保存的权重时，可以通过以下方法完成操作:

loaded_state_dict = torch.load('efficientnet_v2s_imagenet1k_weights.pth')

# 创建未初始化权重的新模型实例
new_model = models.efficientnet_v2_s()

# 载入之前保存的状态字典
new_model.load_state_dict(loaded_state_dict)