from local

This commit is contained in:
dohe0342 2022-12-10 14:24:44 +09:00
parent 226d180832
commit 46090c08b1
3 changed files with 1 additions and 0 deletions

View File

@ -201,6 +201,7 @@ class ScaledAdam(BatchedOptimizer):
)
super(ScaledAdam, self).__init__(params, defaults)
print('fuck!@!!!', len(self.param_groups))
assert len(self.param_groups) == len(parameters_names)
self.parameters_names = parameters_names
self.show_dominant_parameters = show_dominant_parameters