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:
how could i implement the first option you offered, i think it would be the better approach out of 3.
