可能是版本问题? 在目录中 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'
我正在定义参数