What does the register_hook do? #87
Description
Hello, what does register_hook do in byol_pytorch? I get an AssertionError:hidden layer(avgpool) not found, thanks
{{ message }}
Hello, what does register_hook do in byol_pytorch? I get an AssertionError:hidden layer(avgpool) not found, thanks