5f3cddb62a332dc7d98dfacebf94d10bf7874600,arch/example.py,CNN,__init__,#CNN#,22

Before Change

        assert len(block_params) + 1 == len(features)

        blocks = [HighwayBlock(features[i], features[i + 1], **common_block_params, **block_params[i]) for i in range(len(block_params))]
        self.blocks = torch.nn.Sequential(*blocks)

    def forward(self, inp):  // pylint: disable=W
        """
        :param inp: [batch, features, x, y, z]

After Change

        blocks = [HighwayBlock(features[i], features[i + 1], **common_block_params, **block_params[i]) for i in range(len(block_params))]

        self.sequence = torch.nn.Sequential(
            *blocks,
            AvgSpacial(),
            torch.nn.Linear(20, 50),
            torch.nn.ReLU(),
            torch.nn.Linear(50, 2),
        )

    def forward(self, inp):  // pylint: disable=W
        """
        :param inp: [batch, features, x, y, z]

Instances