- 博客(1)
- 资源 (5)
- 收藏
- 关注
原创 PyTorch的hook及其在Grad-CAM中的应用
文章目录hook简介PyTorch的四个hook1. torch.Tensor.register_hook(hook)2. torch.nn.Module.register_forward_hook3. torch.nn.Module.register_forward_pre_hook4.torch.nn.Module.register_backward_hookhook简介pytorch中的...
2019-07-30 21:54:32 7891 6
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人