bvanaken/clinical-assertion-data

saving and loading the model

Opened this issue · 1 comments

Hi,
i was able to save the model using torch.save(model, 'path/to/model/file.pt')
however, when loading the model using torch.load('path/to/model/file.pt') give the error
framework, model = infer_framework_load_model(model, config=model.config)
AttributeError: module 'model' has no attribute 'config'

I was just wondering if you it is possible to save the model on the disk?
Thanks

Hi,
have you further fine-tuned the model from the huggingface hub (https://huggingface.co/bvanaken/clinical-assertion-negation-bert) or do you just want to save the hub checkpoint locally?

If you further trained it, I think I would need a bit more context about which setup you used for that. Could you provide the rest of the script for that?
Best regards
Betty