lightwood icon indicating copy to clipboard operation
lightwood copied to clipboard

Allow to `finetune` GPU-trained models on CPU

Open StpMax opened this issue 1 year ago • 1 comments

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.

StpMax avatar Aug 18 '23 12:08 StpMax

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

yuhuishi-convect avatar Aug 27 '23 23:08 yuhuishi-convect

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

paxcema avatar Aug 09 '24 23:08 paxcema