thanks for the quick reply,
im kind of new to the Pytorch framework, still trying to get a grasp on it.
thats the model class:

image.png1008×668 37.1 KB

how could i implement the first option you offered, i think it would be the better approach out of 3.