Update unet_wo_t/DDPM_Unet_wo_t.py
Browse files
unet_wo_t/DDPM_Unet_wo_t.py
CHANGED
@@ -851,8 +851,8 @@ class Unet(nn.Module):
|
|
851 |
|
852 |
|
853 |
# output dir
|
854 |
-
save_path = 'unet_wo_t/model'
|
855 |
-
log_path = 'unet_wo_t/log'
|
856 |
|
857 |
if not os.path.exists(log_path):
|
858 |
os.mkdir(log_path)
|
|
|
851 |
|
852 |
|
853 |
# output dir
|
854 |
+
save_path = '/content/DDPM_ResNet_Unet/unet_wo_t/model'
|
855 |
+
log_path = '/content/DDPM_ResNet_Unet/unet_wo_t/log'
|
856 |
|
857 |
if not os.path.exists(log_path):
|
858 |
os.mkdir(log_path)
|