tensorboardx profiler execution failed解决

add_graph时报错profiler execution failed,但并没有打印详细的信息,在tensorboardx包的pytorch_graph.py中找到print('profiler execution failed')这行,添加打印RuntimeError。

你可能感兴趣的:(深度学习,pytorch,深度学习,tensorboardx)