Shu-Yu Huang
Apr 22, 2022

--

retain_graph=True 可以把graph保留一次

若要三個backward都同時算到的話,除最後一個backward以外都要做retain_graph=True

--

--

Shu-Yu Huang

AI engineer in Taiwan AI Academy| Former process engineer in TSMC| Former Research Assistance in NYMU| Studying few-shot-learning and GAN