Comments (3)
mrdbourke
commented on September 26, 2024
5
Hi @jdrp ,
Looks like a few people are getting this issue with the latest version of torchvision.
You can find a solution here: #693 (reply in thread)
In short,
This should help fix it:
Overriding torchvision.models._api.WeightsEnum.get_state_dict() should do the trick, LMK if it doesn't work
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights
from torchvision.models._api import WeightsEnum
from torch.hub import load_state_dict_from_url
def get_state_dict(self, *args, **kwargs):
kwargs.pop("check_hash")
return load_state_dict_from_url(self.url, *args, **kwargs)
WeightsEnum.get_state_dict = get_state_dict
efficientnet_b0(weights=EfficientNet_B0_Weights.IMAGENET1K_V1)
efficientnet_b0(weights="DEFAULT")Source: pytorch/vision#7744 (comment)
from pytorch-deep-learning.
jdrp
commented on September 26, 2024
That worked, thanks for the quick help!
from pytorch-deep-learning.
Aconsumeeverything
commented on September 26, 2024
that worked thank you !!
from pytorch-deep-learning.
Related Issues (20)
- Expanding to RNN and Transformer? HOT 1
- Installing Gradio as gd in PyTorch , issues with typing extensions ....
- TypeError: 'Tensor' object is not callable HOT 1
- Getting very less accuracy for Image HOT 1
- Unable to get Reproducibilty of Tensors on GPU | Exercises | PyTorch Fundamentals HOT 3
- Different inputs on Conv2d HOT 3
- 'tuple' object is not callable HOT 1
- 06-pytorch-transfer-learning: got some error in [3.2 Setting up a pretrained model] (torch:2.1.0+cu121, torchvision:0.16.0+cu121) HOT 3
- transformer encoder layer output deos not seem right!
- Broken Link in PyTorch Custom Datasets Notebook HOT 1
- Hey i just reviewed one part of code.
- HuggingFace Spaces Runtime Error
- Getting Started with Tensor flor - scalar.ndim HOT 1
- an error on Batching FashionMNIST figure
- TypeError: 'Tensor' object is not callable HOT 2
- Fix Code Error
- Less accuracy on local machine compare to Colab HOT 1
- The Markdown table formatting is incorrect HOT 1
- PyTorch
- 09 Model development: TypeError: AsyncConnectionPool.__init__() got an unexpected keyword argument 'socket_options'
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from pytorch-deep-learning.