pytorch
qq_42017498
这个作者很懒,什么都没留下…
展开
-
pytorch网络结构可视化
一、import torchimport torchvision.modelsimport hiddenlayer as hlmodel = VGG()##VGG换成对应的自己的网络结构,但是前面需要全部的网络结构hl.build_graph(model, torch.zeros([1, 3,480, 480]))#自定义一个初始输入二、model = VGG()x = to...原创 2019-10-03 11:33:32 · 614 阅读 · 0 评论 -
pytorch nn.upsample
https://www.cnblogs.com/wanghui-garcia/p/11399053.html转载 2019-10-03 15:23:18 · 739 阅读 · 0 评论