pytorch_modelsize

pytorch_modelsize copied to clipboard

Fails with nn.ModuleList

1

I have a model defined with a sequence of layers stored in a nn.ModuleList. For example I find it useful for defining a fully connected neural network (MLP) where you...

alex-golts

not work with conv3d

somebodyus

The original implementation of this tool relies on accessing tensor operations through `model.modules()`. This is simple, but cannot account for arbitrary dimensionality changes in the `.forward()` method. For instance, it's...

jacobkimmel

Half precision implementation

1

Hi Jacob, I tried to estimate the size of my model that uses half precision without any luck. I changed line 34 to HalfTensor, but got the following message: RuntimeError:...

je-santos