摘要: 1、with torch.no_grad():可以用来节省内存,在不需要反向传播的时候,特别是evaluate net的时候,这个时候的net因为 是train过的,所以是requires_grad=True,如果为了节省内存,忽略里面存的梯度和计算图什么的,就可以使用这个; 2、model_cop 阅读全文
posted @ 2019-06-06 13:33 loooffeeeey 阅读(144) 评论(0) 推荐(0)