Hi Alex
Sorry for late replying back.
m.mmmm you are right.
Actually I realized that I did not normalized my data. Do you have any idea that how I can do it by using data loader module ? I tried to use below code, however I don’t have good feeling about that. I believe that there might be a way of normalizing in pytorch rather than do it by for loops
//////////////////////////////////////////////////////////////////////////////////////////////
#again reshape the data file
new_data = np.empty((256,256,94))
for i in range (256) :
for j in range (256) :
for k in range (94) :
  new_data[i,j,k] = (data[k,i,j]-(np.mean(data[:,i,j])))/((np.amax(data[:,i,j]))-(np.amin([data[:,i,j]])))

////////////////////////////////////////////////////////////////////////////////////////
Thank you