@ptrblck
My “resnet20” is supposed to have in inputs as model(image,extra_scalar) with
images= Nx (5, 64, 64)  and extra_scalar=Nx1
and the outputs are Nx180 (ie a vector of length 180) that ones then use with a F.softmax(out,dim=1) to get probabilities.
Well, to give you some real data and the model trained I do not know how to do it. 100k samples is about 7.7G…