·
5 commits
to main
since this release
Release model weight fine-tuned on CIFAR
This release includes fine-tuned model weights.
you can use model weights using below command
import torch
model_name = 'efficientnet_v2_s'
weight_path = 'efficientnet_v2_s_cifar100.pth'
model = torch.hub.load('hankyul2/EfficientNetV2-pytorch', model_name, nclass=100, skip_validation=True)
model.load_state_dict(torch.load(weight_path))