TorchSummaryX 可视化卷积核,特征图尺寸步骤

我发现好像没有人使用torchsummaryX 功能包进行卷积神经网络的可视化 https://github.com/nmhkahn/torchsummaryX 强烈推荐,最近在看Attention Unet 网络架构,始终不知道卷积核大小和特征图大小。但是安装该工具包之后,还是有点一目了然。 1.安装pytorch. https://pytorch.org/ 在pytorch 官网中找自己的下载
相关文章
相关标签/搜索