lucidrains/byol-pytorch

What does the register_hook do?

YTHmamba opened this issue · 0 comments

YTHmamba commented

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