Allow to `finetune` GPU-trained models on CPU

mindsdb / lightwood

Lightwood is Legos for Machine Learning.

GNU General Public License v3.0

450 stars 94 forks source link

Allow to `finetune` GPU-trained models on CPU #1187

Closed StpMax closed 3 months ago

StpMax commented 1 year ago

At the moment model that trained on GPU can be finetuned only on GPU. Will be good to be able make it on CPU.

Currently error is:

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

yuhuishi-convect commented 1 year ago

Seems like duplicates of https://github.com/mindsdb/mindsdb/issues/6949

paxcema commented 3 months ago

Closing (solved in mindsdb, here it is a duplicate of #1129)