Pytorch tensorboardX 网络结构可视化报错

最新推荐文章于 2024-07-22 22:02:52 发布

城俊BLOG

最新推荐文章于 2024-07-22 22:02:52 发布

阅读量3.1k

点赞数

分类专栏： Python # DL-基础 # CV-基础

本文链接：https://blog.csdn.net/qxqxqzzz/article/details/102480812

版权

Python 同时被 3 个专栏收录

307 篇文章 38 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

CV-基础

107 篇文章 2 订阅

订阅专栏

DL-基础

83 篇文章 2 订阅

订阅专栏

在使用tensorboardX可视化Pytorch模型网络结构时遇到错误，问题可能与tensorboardX版本有关。解决方法包括：确保使用1.7版本的tensorboardX；在模型放置到GPU前进行可视化；网络输出需为tuple或tensor而非list。通过这些调整，可以成功解决报错问题。

摘要由CSDN通过智能技术生成

5，使用tensorboardX可视化模型网络结构，各种报错：

报错1：

TypeError: __init__() got an unexpected keyword argument 'record_shapes'

原因：可能需要使用1.7版本的tensorboardX

报错2-6：

RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradient

TypeError: expected sequence object with len >= 0 or a single integer

TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

ValueError: only one element tensors can be converted to Python scalars

RuntimeError: Cannot encode

了解本专栏