The issue is that pytorch does not require a models input layer to specify a dimension or size, only amounts of in features and out features (and kernel size maybe). You always have the option of looping over the modules within the script module to see these parameters, you just won’t see dimensions unless they are put in through customs modules. Models will not store this information unless explicitly made to do so