ndbao2002 commited on
Commit
04469b8
1 Parent(s): 283987a

Update unet_wo_t/DDPM_Unet_wo_t.py

Browse files
Files changed (1) hide show
  1. unet_wo_t/DDPM_Unet_wo_t.py +2 -2
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)