pytorch可视化
wwlCape
这个作者很懒,什么都没留下…
展开
-
pytorch 打印参数
**遍历named_parameters()中的所有的参数,只打印那些param.requires_grad=True的变量**for name, param in model.named_parameters(): if param.requires_grad: print(name)若想可视化某一层的参数,可以先转列表,然后索引 params = list...原创 2019-11-12 20:32:37 · 479 阅读 · 0 评论 -
tensorboardx当网络模型输入两个参数时add_graph()报错解决方法
pytorch:0.4.0python:3.6File "/user/files/apps/anaconda2/envs/pytorch/lib/python3.6/site-packages/tensorboardX/graph.py", line 85, in graphlist_of_nodes = parse(graph)File "/user/files/apps/anacond...原创 2019-11-09 16:47:31 · 3234 阅读 · 0 评论