Function conv3x3

config/atari/model.py:52–55

(in_channels, out_channels, stride=1)

def conv3x3(in_channels, out_channels, stride=1):
    return nn.Conv2d(
        in_channels, out_channels, kernel_size=3, stride=stride, padding=1, bias=False
    )


# Residual block