Netron神经网络结构可视化只显示权重没有箭头,已解决
好不容易写好了网络,保存成了.h5格式,放进Netron里作图发现我的图黑黑丑丑的而且只显示权重,如下图:
可是看别人画得花花绿绿的还漂亮,为什么到我这里就剩权重了呢?查了一些办法,还废了好大一把劲儿特意把.h5的模型转换成了.onnx格式。结果更是废,造成这种情况的原因,其实不是模型的格式问题,而是因为保存模型时犯了错,我只保存了权重,没有保存完整的网络结构,所以才出现这种情况。
import pandas
pandas.DataFrame(history_callback.history).to_csv("/content/drive/My Drive/AlexNet.csv")