一、验证集评估阶段的张量计算非常占用空间。 法1. 可以对利用torch.tensor().detach().cpu().numpy()转为numpy,在cpu上进行loss和acc的计算 法2. 直接对评估阶段使用with torch.no_grad():
pytroch运行时出现:CUDA error:out of memory
最新推荐文章于 2024-05-24 12:32:34 发布
一、验证集评估阶段的张量计算非常占用空间。 法1. 可以对利用torch.tensor().detach().cpu().numpy()转为numpy,在cpu上进行loss和acc的计算 法2. 直接对评估阶段使用with torch.no_grad():