Ranger-Deep-Learning-Optimizer icon indicating copy to clipboard operation
Ranger-Deep-Learning-Optimizer copied to clipboard

TypeError in GC operation for Conv layers and FC layers

Open ZhWL123456 opened this issue 4 years ago • 0 comments

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, )

ZhWL123456 avatar Jul 17 '20 09:07 ZhWL123456