TypeError in GC operation for Conv layers and FC layers

lessw2020 / Ranger-Deep-Learning-Optimizer

Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase

Apache License 2.0

1.19k stars 176 forks source link

Open ZhWL123456 opened 4 years ago

ZhWL123456 commented 4 years ago

TypeError: mean() received an invalid combination of arguments - got (keepdim=bool, dim=tuple, ), but expected one of:

()
(torch.dtype dtype)
(int dim, torch.dtype dtype) didn't match because some of the keywords were incorrect: keepdim
(int dim, bool keepdim, torch.dtype dtype)
(int dim, bool keepdim) didn't match because some of the arguments have invalid types: (dim=tuple, keepdim=bool, )