Hi ptrblck,
Thanks for help. I was using “list(Model.parameters())” to try to get the parameters but it gave 6 tensors and hence I was confused as to whether the ReLU or the MaxPooling layers had trainable parameters. I tried using your method and “Model.state_dict()” and I was able to get the parameters for the Conv1D, BatchNorm and Conv1DTransposed but however when I use “Model.state_dict()”, my tensors have the “requires_grad” attribute set to False. If I loaded those layers, am I able to load them in as is or do I still have to explicitly load them with “no_grad()” while having codes that set their “requires_grad” attribute to False?