pytorch项目报错:UnboundLocalError: local variable ‘count

pytorch项目报错:UnboundLocalError: local variable ‘count

pytorch项目报错:UnboundLocalError: local variable 'count_module_10' referenced before assignment

• 错误提示：UnboundLocalError: local variable 'count_module_10' referenced before assignment
• 解决办法

错误提示：UnboundLocalError: local variable ‘count_module_10’ referenced before assignment

解决办法

我的解决思路来源于这篇文章python local variable ‘xxx’ referenced before assignment 警告的解决方法
将代码修改这样：