Your assignment might work, but I would always double check, if you really get the desired outputs.
Since you are assigning the features block to pt_model, you assume that this block is executed sequentially in the original forward method, which might not always be the case.
To just reuse the original forward method, you could use:
pt_model.classifier = nn.Identity()

to replace the classifier module with a simple identity method.