关于模型可视化的问题

#1

我想模型可视化，所以按照官网指南将模型存储到.tm文件中，但是按照官网的说法，还有个data作为输入，但我不清楚data从哪里来的，我自己保存的model是训练好的，内部网络结构还是网络参数应该都是有的，所以我写的时候tm.traced_module()里只填了model一个参数，结果运行的时候就显示module object is not callable，可我是按照官网上写的import megengine.traced_module as tm，应该不是导入模块的问题

QQ图片20240105094001
QQ图片20240105094001781×186 11.7 KB
T3OGPN}NI~6KIP%FLGFQF)V_tmb

#2

可以参考例子中给一个随机数生成的输入，另外转为traced_module前需要model.eval()

#3

那一行我注意到了，但是不清楚为什么需要使用随机数 :flushed: :flushed:

#4

这里给的输入data仅仅是建traced_module需要，需要给对shape和dtype，数值是无所谓的，所以例子中使用了随机数值

1赞
#5

我理解原因是：模型trace的时候会把模型跑一遍来转换为静态图，所以必须要指定输入数据。

1赞