search

lucidrains / byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
MIT License
1.76k stars 248 forks source link

What does the register_hook do? #87

Open YTHmamba opened 1 year ago

YTHmamba commented 1 year ago

Hello, what does register_hook do in byol_pytorch? I get an AssertionError:hidden layer(avgpool) not found, thanks