search

jacobkimmel / pytorch_modelsize

Estimates the size of a PyTorch model in memory
MIT License
354 stars 49 forks source link

not work with conv3d #9

Open somebodyus opened 4 years ago

somebodyus commented 4 years ago

not work with conv3d