retain_graph
时间: 2023-09-20 12:05:51 浏览: 117
Pytorch 中retain_graph的用法详解
retain_graph参数是指在调用backward函数时,是否保留计算图。如果设置为True,则计算图将被保留,可以多次使用backward函数进行梯度计算。如果设置为False,则在计算完梯度后,计算图将被释放以节省空间。在大多数情况下,不需要设置retain_graph为True,并且通常可以通过更高效的方式来处理。但是,在需要进行多次backward且存在共享计算图的情况下,需要先设置retain_graph为True,然后再进行梯度计算。如果不设置retain_graph为True并尝试第二次进行backward,可能会报错提示"Trying to backward through the graph a second time, but the buffers have already been freed."
阅读全文