Pycharm complains about using [] operator with pytorch's nn.ModuleList object

  • Thread starter Michael Heidelberg
  • Start date
M

Michael Heidelberg

Guest
Pycharm's type-checking complains about the following block of code (taken from pytorch's docs):

import torch.nn as nn
class MyModule(nn.Module):
def __init__(self):
super(MyModule, self).__init__()
self.linears = nn.ModuleList([nn.Linear(10, 10) for i in range(10)])

def forward(self, x):
# ModuleList can act as an iterable, or be indexed using ints
for i, l in enumerate(self.linears):
x = self.linears[i // 2](x) + l(x)
# ^---------- complains here about:
# Cannot find reference '[' in 'input:(Any, ...), kwargs:dict) -> Any'
return x


Is there a way to fix this issue?

Continue reading...
 
Top