update: OPTIMIZER_NAMES

kozistr · kozistr · commit fa05118687de · 2022-02-01T13:52:04.000+09:00
diff --git a/tests/test_optimizer_parameters.py b/tests/test_optimizer_parameters.py
@@ -30,6 +30,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
     'diffrgrad',
     'lamb',
     'ralamb',
+    'lars',
 ]
 
 BETA_OPTIMIZER_NAMES: List[str] = [

Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:`
`30`	`30`	`'diffrgrad',`
`31`	`31`	`'lamb',`
`32`	`32`	`'ralamb',`
	`33`	`+ 'lars',`
`33`	`34`	`]`
`34`	`35`
`35`	`36`	`BETA_OPTIMIZER_NAMES: List[str] = [`