Solved by sophia_k in post #5 
              
              
                UPDATE: Error Solved! 
I found the cause of the error: 
My fault, I put a decorator from dataclasses in front of the BaseModel, and this is what caused the issue. 
@dataclass
class BaseRGCNHetero(nn.Module):
...

So for everybody else having this issue, just do not use the @dataclass decorator for m…