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

Before Change

        ]
        common_block_params = {"size": 4, "n_radial": 1, "stride": 2, "batch_norm_before_conv": False}
        block_params = [
            {"non_linearities": True},
            {"non_linearities": True},
            {"non_linearities": False},
        ]

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

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