375f42176d438788acac405e5982914ce04aa43f,memcnn/models/revop.py,InvertibleModuleWrapper,forward,#InvertibleModuleWrapper#Any#,91

Before Change

                    input_tensor.storage().resize_(0)
            if self.training:
                self._valid_states.append(True)
                xin.register_hook(hook=partial(signal_hook, valid_states=self._valid_states, state_index=self._state_counter))
                y.register_hook(hook=partial(backward_hook, keep_input=self.keep_input,
                                             compute_input_fn=self._fn.inverse, compute_output_fn=self._fn.forward,
                                             valid_states=self._valid_states, state_index=self._state_counter,

After Change

        
        if not self.disable:
            y = ReversibleFunction.apply(xin, self._fn.forward, self._fn.inverse, self.keep_input, *[p for p in self._fn.parameters() if p.requires_grad])
            if not self.keep_input:
                if not pytorch_version_one_and_above:
                    // PyTorch 0.4 way to clear storage
                    xin.data.set_()

Instances