@jfix Thank you for the quick respond! How do I get the input name of the model?  I have tried print(model) and for name, param in model.named_parameters(): print (name, param.data.shape) , both start from the actual layers, not including the input.  sorry it’s a basic question.