resnet18_weights下载
时间: 2023-11-16 09:55:09 浏览: 35
很抱歉,我无法提供您所需的资源下载链接。但是,您可以在以下网站中找到ResNet-18的预训练权重:
- PyTorch官方模型库:https://pytorch.org/docs/stable/torchvision/models.html
- GitHub上的模型库:https://github.com/pytorch/vision/tree/master/torchvision/models
相关问题
Arguments other than a weight enum or None for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing weights=ResNet18_Weights.IMAGENET1K_V1. You can also use weights=ResNet18_Weights.DEFAULT to get the most up-to-date weights. warnings.warn(msg)解决办法解决办法示例
以下是解决这个问题的示例代码:
```python
from torchvision.models import resnet18
from torchvision.models.resnet import ResNet, BasicBlock, Bottleneck, model_urls
class ResNet18(ResNet):
def __init__(self, pretrained=True):
super(ResNet18, self).__init__(BasicBlock, [2, 2, 2, 2])
if pretrained:
# 使用预训练的模型权重
self.load_state_dict(model_zoo.load_url(model_urls['resnet18']))
model = ResNet18(weights=None) # 或者使用 weights=ResNet18_Weights.DEFAULT
```
这里的解决方法是将传入的`weights`参数值改为`None`或`ResNet18_Weights.DEFAULT`。同时,建议在代码中加上以下语句以忽略该警告:
```python
import warnings
warnings.filterwarnings('ignore')
```
AttributeError: module itorchvision.models' has no attribute 'ResNet18_Weights
在您的问题中,您遇到了一个AttributeError: module 'torchvision.models' has no attribute 'ResNet18_Weights'的错误。这意味着torchvision.models模块中没有名为'ResNet18_Weights'的属性。
要解决这个问题,您可以尝试以下方法之一:
1. 检查torchvision的版本:确保您使用的是最新版本的torchvision。您可以使用pip install torchvision --upgrade命令来更新torchvision到最新版本。
2. 检查模型名称:确保您使用的是正确的模型名称。ResNet18_Weights可能不是torchvision.models中可用的模型之一。您可以查看torchvision.models模块的文档,以确定可用的模型名称。
3. 自定义模型:如果您要使用的模型不是torchvision中提供的预训练模型之一,您可以考虑自定义一个模型。您可以使用torch.nn模块来构建自己的模型。
请注意,在上述解决方法中,第一个方法是最常见的解决方法,确保您使用的是最新版本的torchvision,并查看可用的模型名称。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [AttributeError: module ‘torchvision.models‘ has no attribute ‘xxxx‘ 问题和解决](https://blog.csdn.net/Davidietop/article/details/122296013)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 33.333333333333336%"]
- *2* [AttributeError: module 'tensorflow.compat.v1' has no attribute '](https://download.csdn.net/download/qq_38766019/86272235)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 33.333333333333336%"]
- *3* [AttributeError: module ‘torchvision.models‘ has no attribute ‘regnet_y_400mf](https://blog.csdn.net/weixin_41287060/article/details/129798585)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 33.333333333333336%"]
[ .reference_list ]