在 colab 上收到错误消息出了什么问题?

问题描述 投票:0回答:0

可能是版本问题? 在目录中 build_fastai_model() 可用 代码:

input_shape = [224, 224]
batch_size = 32
num_examples = -1
device=  "cuda"
epochs = 100
plot_freq = 5

model = build_fastai_model()

optimizer = torch.optim.Adam(model.parameters(), lr=1e-4)
loss_fn = nn.MSELoss()

错误消息:

AttributeError                            Traceback (most recent call last)
<ipython-input-25-3211e656934c> in <module>
      6 plot_freq = 5
      7 
----> 8 model = build_fastai_model()
      9 
     10 optimizer = torch.optim.Adam(model.parameters(), lr=1e-4)

3 frames
/usr/local/lib/python3.8/dist-packages/fastai/vision/learner.py in _get_first_layer(m)
     32     "Access first layer of a model"
     33     c,p,n = m,None,None  # child, parent, name
---> 34     for n in next(m.named_parameters())[0].split('.')[:-1]:
     35         p,c=c,getattr(c,n)
     36     return c,p,n
AttributeError: 'function' object has no attribute 'named_parameters'

我正在定义参数

image-processing deep-learning computer-vision resnet fast-ai
© www.soinside.com 2019 - 2024. All rights reserved.