python包:torchsummary

利用torchsummary觀察每一層的情況

 

1)按照方式   

pip install torchsummary

 

 

2)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章