Allow to `finetune` GPU-trained models on CPU

mindsdb / lightwood

Lightwood is Legos for Machine Learning.

GNU General Public License v3.0

434 stars 92 forks source link

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

Open StpMax opened 10 months ago

StpMax commented 10 months 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 10 months ago

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