1b8b1ee7ac530f74a31a72cd28add5ad5ec37c45,memcnn/models/revop.py,ReversibleModule,forward,#ReversibleModule#,89

Before Change

                    // PyTorch 1.0+ way to clear storage
                    input_tensor.storage().resize_(0)
            if self.training:
                xin.register_hook(hook=partial(signal_hook, rev_block=self, direction="forward"))
                y.register_hook(hook=partial(backward_hook, keep_input=self.keep_input, rev_block=self,
                                             compute_input_fn=self._fn.inverse, compute_output_fn=self._fn.forward,
                                             direction="forward", input_tensor=input_tensor, output_tensor=output_tensor))

After 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,

Instances