def forward()

in ppo_ewma/impala_cnn.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def forward(self, x):
        x = self.firstconv(x)
        if getattr(self, "pool", True):
            x = F.max_pool2d(x, kernel_size=3, stride=2, padding=1)
        for block in self.blocks:
            x = block(x)
        return x