from local

This commit is contained in:
dohe0342 2023-04-11 16:05:10 +09:00
parent ecb581d7db
commit 91eb32b1e6
3 changed files with 4 additions and 0 deletions

View File

@ -81,6 +81,10 @@ class BatchedOptimizer(Optimizer):
list list
) # `batches` maps from tuple (dtype_as_str,*shape) to list of str ) # `batches` maps from tuple (dtype_as_str,*shape) to list of str
print(len(param_group))
print(len(group_params_names))
exit()
assert len(param_group) == len(group_params_names) assert len(param_group) == len(group_params_names)
for p, named_p in zip(param_group, group_params_names): for p, named_p in zip(param_group, group_params_names):
key = (str(p.dtype), *p.shape) key = (str(p.dtype), *p.shape)