from paddle.vision.models import resnet18
# build model
model = resnet18(pretrained=True, num_classes=102, with_pool=True)
from paddle.metric import Accuracy
model = paddle.Model(model)
model.load('./work/final') # 将保存的参数加载到当前的模型中
model.fit(train_dataset,
epochs=1,
batch_size=32,
verbose=1,
save_dir='/home/aistudio/work',
callbacks=callback
)
-17-0c970b3f9b0b> in
7 save_dir='/home/aistudio/work',
8 #places='/home/aistudio/work',
----> 9 callbacks=callback
10 )
11 #model.save('/home/aistudio/work',False)
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/hapi/model.py in fit(self, train_data, eval_data, batch_size, epochs, eval_freq, log_freq, save_dir, save_freq, verbose, drop_last, shuffle, num_workers, callbacks)
train_data,
1673 batch_sampler=train_sampler,
-> 1674 places=self._place,
1675 num_workers=num_workers,
1676 return_list=True)
AttributeError: 'Model' object has no attribute '_place'
所以callback是哪里来的?
为啥啊?
callback = paddle.callbacks.VisualDL(log_dir='visualdl_log_dir')
model.fit(train_dataset,
epochs=1,
batch_size=32,
verbose=1,
save_dir='/home/aistudio/work',
callbacks=callback
)
加上了,还是报同样的错误。
你把这个注释掉试试,是不是就跑的通了?