LOADING . . .

2023年10月7日
pytorch的四个hook函数
摘要： 训练神经网络模型有时需要观察模型内部模块的输入输出，或是期望在不修改原始模块结构的情况下调整中间模块的输出，pytorch可以用hook回调函数来实现这一功能。主要使用四个hook注册函数：register_forward_hook、register_forward_pre_hook、registe 阅读全文
posted @ 2023-10-07 16:04 颀周 阅读(318) 评论(0) 推荐(0) 编辑
很高兴能帮到你~
点赞