I got the root cause of the problem.
my target tensor is containing values like 15 , 40 etc.
x,y=train_dataset.getitem(0)
y contains values like 15,40 etc.
So , can you please help me in converting image tensor to 0,1 ?
Thanks in advance.