您是否需要设置 `retain_graph=True`?
作者: tengerye创建于 2024年12月6日更新于 2024年12月6日
你好,我认为你应该使用 retain_graph=True 这里,否则,图形将从这里被释放,而值函数的更新将不会影响行为者的更新。
内容来源: lucidrains/PaLM-rlhf-pytorch
你好,我认为你应该使用 retain_graph=True 这里,否则,图形将从这里被释放,而值函数的更新将不会影响行为者的更新。
内容来源: lucidrains/PaLM-rlhf-pytorch