TypeError in GC operation for Conv layers and FC layers

[ZhWL123456]

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