import VisualizeGraph as viz

x = Variable(torch.randn(1, 3, args.inWidth, args.inHeight))    # 产生随机X

if args.onGPU == True:
            x = x.cuda()

y = model.forward(x)                #传入网络模型
g = viz.make_dot(y)                 #可视化有向图g
g.render(args.savedir + '/model.png', view=False)    #保存有向图 view =False代表运行时不弹出图片

 

# python
Logo
昇腾开源生态专区

昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐

cover

GLM-4.7-Flash模型在FP16精度下部署需求

avatar 昇腾开源生态专区

AReaL x 昇腾，加速大模型全异步RL训练创新

AReaL框架为需要在昇腾平台进行强化学习的开发者提供了新的可靠途径——开箱即用保障开发者轻松上手，优秀架构支撑模型性能。AReaL框架在昇腾平台上会持续演进，为开发者提供更强大、更便捷的强化学习体验，大家可以持续关注AReaL开源项目了解最新的技术动态。AReaL开源项目：https://github.com/inclusionAI/AReaL华为云深耕大模型后训练全链路优化，特别是在 NPU

avatar 昇腾开源生态专区

源码交付破局异构算力：基于GB28181/RTSP与Docker的AI视频平台架构实战

在安防行业深耕十年的架构师生涯中，我见证了无数技术团队在基础设施的泥潭中挣扎。传统的视频监控项目交付往往面临三大“拦路虎”：协议碎片化：海康、大华、宇视等厂商私有协议林立，GB28181国标对接调试周期长，RTSP/RTMP流媒体服务在高并发下稳定性难以保障。异构算力适配难：客户现场环境极其复杂，既有x86架构的NVIDIA GPU服务器，又有ARM架构的华为昇腾、瑞芯微NPU边缘盒子。为不同芯片

avatar 昇腾开源生态专区