Closed kabouzeid closed 2 years ago
Why not l.register_forward_hook(self._register_hs_hook) here?
l.register_forward_hook(self._register_hs_hook)
https://github.com/Guillem96/data2vec-vision/blob/f183fa6ebf84eafe723ee2c6c6a5b48c2b5809fb/data2vec_vision/model.py#L55
Yeah, probably it is neater. Feel free to open a PR.
Why not
l.register_forward_hook(self._register_hs_hook)
here?https://github.com/Guillem96/data2vec-vision/blob/f183fa6ebf84eafe723ee2c6c6a5b48c2b5809fb/data2vec_vision/model.py#L55