search

Guillem96 / data2vec-vision

PyTorch implementation of Data2Vec self-supervised approach for vision use cases.
GNU General Public License v3.0
18 stars 5 forks source link

Question #2

Closed kabouzeid closed 2 years ago

kabouzeid commented 2 years ago

Why not l.register_forward_hook(self._register_hs_hook) here?

https://github.com/Guillem96/data2vec-vision/blob/f183fa6ebf84eafe723ee2c6c6a5b48c2b5809fb/data2vec_vision/model.py#L55

Guillem96 commented 2 years ago

Yeah, probably it is neater. Feel free to open a PR.