function grad.mean() Error

[lucaskyle]

sorry I got an Error in optimizer GCC

grad.add_(-grad.mean(dim = tuple(range(1,len(list(grad.size())))), keepdim = True))

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

• no arguments
• (int dim)
• (int dim, bool keepdim) didn't match because some of the arguments have invalid types: (dim=tuple, keepdim=bool, )

dim should not be tuple right? my pytorch vision 0.3.1

[lucaskyle]

sorry for bothering u . i fixed it. just use pytorch upon 1.10.