CH.Li

@shivam-aggarwal maybe you can try to replace the `clear()` method to `pop()` in function `NetWrapper.get_representation()` looks like: ``` def get_representation(self, x): if self.layer == -1: return self.net(x) if not self.hook_registered:...