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

@ -80,6 +80,10 @@ class BatchedOptimizer(Optimizer):
batches_names = defaultdict(
list
) # `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)
for p, named_p in zip(param_group, group_params_names):