From af3e110dcf5a0495306d9b71eb93436920360af4 Mon Sep 17 00:00:00 2001
From: Tobias Seibel <55710042+SeibelT@users.noreply.github.com>
Date: Fri, 23 Jun 2023 14:17:03 +0200
Subject: [PATCH] storing imagetensor and normalized images
---
models/all_unets.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/models/all_unets.py b/models/all_unets.py
index 9fe4f6b..fabd628 100644
--- a/models/all_unets.py
+++ b/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)
+
+
--
GitLab