python包：torchsummary

利用torchsummary观察每一层的情况

 

1）按照方式   

pip install torchsummary

 

 

2）

posted @ 2024-05-04 08:55  wenluderen  阅读(93)  评论(2)    收藏  举报