Thanks for the prompt response. Do you use data augmentation? Sorry for all the questions which could maybe be answered by looking at the code, but I am just getting started with pytorch etc and I am trying to understand the exact input format.