finetune-anything icon indicating copy to clipboard operation
finetune-anything copied to clipboard

如何调整训练使用的GPU的id

Open jifeng35 opened this issue 5 months ago • 0 comments

我已经修改了SamRunner里面的在.cuda时候的gpu_id,以及get_model时候的gpu_id,仍然提示我有一部分参数or数据在cuda:0上 我应当如何解决

jifeng35 avatar Jan 10 '24 12:01 jifeng35