From e54709f20df0877a92ea1d16130190384e74aab4 Mon Sep 17 00:00:00 2001 From: kaanaksit Date: Mon, 13 Jan 2025 21:12:19 +0000 Subject: [PATCH] Improvement for a unit test: test/test_learn_tools_freeze_unfreeze.py --- test/test_learn_tools_freeze_unfreeze.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_learn_tools_freeze_unfreeze.py b/test/test_learn_tools_freeze_unfreeze.py index 51c3b7f9..346c1686 100644 --- 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(