Ranger-Deep-Learning-Optimizer
Ranger-Deep-Learning-Optimizer copied to clipboard
TypeError in GC operation for Conv layers and FC layers
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, )