storing imagetensor and normalized images

models/all_unets.py

@@ -69,7 +69,7 @@ class UNet_Res(nn.Module):
     # first two conv layers
     x = self.first_conv(input) + t_emb0[:,:,None,None]
     #timemb
-    skip1 =x
+    
     skip1,x = self.down1(x,t_emb1)
     skip2,x = self.down2(x,t_emb2)
     skip3,x = self.down3(x,t_emb3)
@@ -238,3 +238,5 @@ class MidBlock_Res(nn.Module):
     x = self.convblock1(x,t)
     return self.convblock2(x,t)
 
+
+