![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
不会起昵称啊
这个作者很懒,什么都没留下…
展开
-
torchsample
pytorch像Keras一样的方式运行模型 from torchsample.modules import ModuleTrainer class Network(nn.Module): pass model = Network() trainer = ModuleTrainer(model) trainer.compile(.......)原创 2020-06-14 13:18:00 · 1370 阅读 · 1 评论 -
pytorch中的钩子Hook)作用
pytorch对于中间变量,一旦它们完成了自身反传的使命,就会被释放掉。 因此,hook就派上用场了。简而言之,register_hook的作用是,当反传时,除了完成原有的反传,额外多完成一些任务。你可以定义一个中间变量的hook,将它的grad值打印出来,当然你也可以定义一个全局列表,将每次的grad值添加到里面去。 至于register_forward_hook和register_backward_hook的用法和这个大同小异。只不过对象从Variable改成了你自己定义的nn.Modu...原创 2020-06-14 13:12:47 · 241 阅读 · 0 评论