平凡之上
博观而约取,厚积而薄发! 守得云开见月明。

1、可以减少内存使用

2、训练集训练好模型后,在验证集这里使用with torch.no_grad(),训练集则不会计算梯度值,然后并不会改变模型的参数,只是看了训练的效果。

posted on 2021-12-02 14:55  平凡之上  阅读(2521)  评论(0编辑  收藏  举报