Pycharm жалуется на использование оператора [] с объектом ntor.ModuleList в pytorch - PullRequest
3 голосов
/ 06 октября 2019

Проверка типа Pycharm жалуется на следующий блок кода (взят из документации pytorch ):

    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

Есть ли способ исправить эту проблему, не подавляя ее?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...