finetune-anything
finetune-anything copied to clipboard
如何调整训练使用的GPU的id
我已经修改了SamRunner里面的在.cuda时候的gpu_id,以及get_model时候的gpu_id,仍然提示我有一部分参数or数据在cuda:0上 我应当如何解决
我已经修改了SamRunner里面的在.cuda时候的gpu_id,以及get_model时候的gpu_id,仍然提示我有一部分参数or数据在cuda:0上 我应当如何解决