Hi, is there anyway to specify the absolute path of where to save models in the SaveModelCallback function? I don’t want to save my model in the data directory… Thanks!
1 Like
learn = create_cnn(data,models.resnet18, metrics=accuracy,
path=Path('D:/models/Translator/'),
callback_fns=[partial(SaveModelCallback,every='epoch',name='1')]).
path
is setting where to save the models and SaveModelCallback
is saying to save the model after every epoch. It’s from this notebook if you want to look
2 Likes