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,
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