Skip to content

Commit ee3ea6c

Browse files
committed
update: test_size_of_parameter
1 parent bdc48b1 commit ee3ea6c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_optimizer_parameters.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,7 @@ def test_ranger21_warm_methods():
223223
@pytest.mark.parametrize('optimizer', ['ranger21', 'adai'])
224224
def test_size_of_parameter(optimizer):
225225
param = simple_parameter(require_grad=False)
226+
param.grad = None
226227

227228
with pytest.raises(ZeroParameterSizeError):
228229
load_optimizer(optimizer)([param], 1).step()

0 commit comments

Comments
 (0)