错误AssertionError: size of input tensor and input format are different

在写代码的过程中出现了如下错误:
AssertionError: size of input tensor and input format are different。

根据查询资料发现是tensorboardX的版本问题。原本安装的是2.0版本,把2.0版本卸载,安装1.4版本即可解决该问题。
cmd卸载指令:
pip uninstall tensorboardX

安装指令
pip install tensorboardX==1.4

猜你喜欢

转载自blog.csdn.net/weixin_41963310/article/details/109046691