from torchsummary import summary
a = summary(eegnet,input_size=(a,b,c),batch_size=-1)
torch 查看网络参数量
最新推荐文章于 2025-03-08 15:53:41 发布
这篇博客介绍了如何利用torchsummary库来对EEGNet模型进行输入尺寸为(a, b, c)的摘要可视化,批量大小设为-1,以便更好地理解和优化模型结构。
4万+

被折叠的 条评论
为什么被折叠?



