ndbao2002
/

DDPM_ResNet_Unet

Model card Files Files and versions

ndbao2002 commited on Mar 26, 2024

Commit

44fbcef

·

verified ·

1 Parent(s): e622aef

Update unet/DDPM_Unet.py

Matched with google colab!

Files changed (1) hide show

unet/DDPM_Unet.py +2 -2

unet/DDPM_Unet.py CHANGED Viewed

@@ -851,8 +851,8 @@ class Unet(nn.Module):
 # output dir
-save_path = 'unet/model'
-log_path = 'unet/log'
 if not os.path.exists(log_path):
     os.mkdir(log_path)

 # output dir
+save_path = '/content/DDPM_ResNet_Unet/unet/model'
+log_path = '/content/DDPM_ResNet_Unet/unet/log'
 if not os.path.exists(log_path):
     os.mkdir(log_path)