ucbdrive/skipnet

Typo in forward of ResNetRecurrentGateSP

Closed this issue · 1 comments

There is a typo in the forward method of ResNetRecurrentGateSP (could be elsewhere as well, haven't checked) under CIFAR, grob instead of gprob. Wrong values are returned up to the top function of the training loop, but thankfully they are not used.

 for g in range(3):
            for i in range(0 + int(g == 0), self.num_layers[g]):
                ...
                mask, grob = self.control(gate_feature)
                gprobs.append(gprob)
                ...