We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1446f6c commit 7de15fcCopy full SHA for 7de15fc
tests/test_optimizer_parameters.py
@@ -120,19 +120,24 @@ def test_betas(optimizer_name):
120
optimizer = load_optimizer(optimizer_name)
121
122
with pytest.raises(ValueError):
123
- optimizer(None, betas=(-0.1, 0.1))
+ if optimizer_name == 'ranger21':
124
+ optimizer(None, num_iterations=100, betas=(-0.1, 0.1))
125
+ else:
126
+ optimizer(None, betas=(-0.1, 0.1))
127
128
- optimizer(None, betas=(0.1, -0.1))
129
130
+ optimizer(None, num_iterations=100, betas=(0.1, -0.1))
131
132
+ optimizer(None, betas=(0.1, -0.1))
133
134
if optimizer_name == 'adapnm':
135
136
optimizer(None, betas=(0.1, 0.1, -0.1))
137
138
139
def test_reduction():
- model: nn.Module = Example()
- parameters = model.parameters()
140
+ parameters = Example().parameters()
141
optimizer = load_optimizer('adamp')(parameters)
142
143
0 commit comments