YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)
import trace_layer2 as models
import torch
x=torch.randn(1, 3, 224, 224)
state_dict = torch.load('swav_imagenet_layer2.pt', map_location='cpu')
model = models.resnet50w2()
model.load_state_dict(state_dict)
model.eval()
feature = model(x)
traced_model = torch.jit.load('traced_swav_imagenet_layer2.pt', map_location='cpu')
traced_model.eval()
feature = traced_model(x)
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support