From 9850cd196c192b25f79d5ae8201027b8b7eb312b Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Mon, 13 Mar 2023 08:55:03 -0700 Subject: [PATCH] Update autogradyt_tutorial.py Fix #2219 --- beginner_source/introyt/autogradyt_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/introyt/autogradyt_tutorial.py b/beginner_source/introyt/autogradyt_tutorial.py index 19c25c5e81d..1ce49f3cd03 100644 --- a/beginner_source/introyt/autogradyt_tutorial.py +++ b/beginner_source/introyt/autogradyt_tutorial.py @@ -334,7 +334,7 @@ def forward(self, x): print(model.layer2.weight.grad[0][0:10]) -optimizer.zero_grad() +optimizer.zero_grad(set_to_none=False) print(model.layer2.weight.grad[0][0:10])