Mdhvince:

So my question is, is there a way with PyTorch to tell my model, ok you can compute 16 coordinates, but when comparing your output with the ground truth, do not take in consideration the coordinates N. Where N is the 4 additional coordinates which are missing in the dataset.


I don’t really think that’s what you want to do. Instead, you should just remove the last layer of your model (that used to output 16 coordinates) and replace it with a new layer just like it but with 12 outputs.