EfficientNetV2-pytorch

EfficientNetV2-pytorch copied to clipboard

import torch from PIL import Image img = Image.open('hen.jpg') model = torch.hub.load('hankyul2/EfficientNetV2-pytorch', 'efficientnet_v2_s', pretrained=True, nclass=1000) import matplotlib.pyplot as plt from torchvision import transforms transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize([0.485, 0.456, 0.406],...

claire199705

File "/root/.cache/torch/hub/hankyul2_EfficientNetV2-pytorch_master/efficientnetv2/pretrained_weight_loader.py", line 123, in load_npy param.data.copy_(npz_dim_convertor(name, weight.get(name))) RuntimeError: output with shape [256, 1, 3, 3] doesn't match the broadcast shape [256, 256, 3, 3]

ShawKai666