Improvement for a unit test: test/test_learn_tools_freeze_unfreeze.py

kaanaksit · Jan 13, 2025 · e54709f · e54709f
1 parent 2ca65c2
commit e54709f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_learn_tools_freeze_unfreeze.py b/test/test_learn_tools_freeze_unfreeze.py
@@ -5,8 +5,8 @@
 
 def check_parameters_gradient(model):
     print(model.parameters)
-    for parameter_id, parameter in enumerate(model.parameters()):
-        print('ID: {}, Name: {}, Requires grad: {}'.format(parameter_id, parameter.name, parameter.requires_grad))
+    for (fullname, parameter) in model.named_parameters():
+        print('Name: {}, Requires grad: {}'.format(fullname, parameter.requires_grad))
 
 
 def test(