2-图形化model-查看模型的基本结构 torch的module的可视化 from torchview import draw_graph model_graph = draw_graph(model, input_size=(1, 1, 28, 28), save_graph=True, expand_nested=True) 评论 全名: E-Mail: 个人主页: 请输入您的评论. 可以使用维基语法: 订阅评论
评论