python包：torchsummary

利用torchsummary观察每一层的情况

 

1）按照方式   

pip install torchsummary

 

 

2）

posted @ 2024-05-04 08:55  wenluderen  阅读(43)  评论(1)  编辑  收藏  举报