usar
with torch.enable_grad():
en lugar de
with torch.no_grad():
Puede evitar eficazmente la fuga de gradiente causada por problemas de código.
Buenos hábitos de codificación para el aprendizaje profundo
Supongo que te gusta
Origin blog.csdn.net/euzmin/article/details/126307097
Recomendado
Clasificación