vdogmcgee

测试脚本运行的问题

2

你好 , 我下载了13B的姜子牙模型 , 但是出现了这个问题 , 请问是什么原因引起的呢 ?  另外貌似设置成 device='cpu' , 也还是会在gpu加载模型吗?

model.load_state_dict(torch.load(save_path)) corrcoef = test(deving_data, model) print(f"dev_corrcoef: {corrcoef:>4f}") test之前应该需要先调用一下model.eval()的吧