Update resnet/DDPM_ResNet.py
Browse files- resnet/DDPM_ResNet.py +2 -2
resnet/DDPM_ResNet.py
CHANGED
|
@@ -773,8 +773,8 @@ class EDSR(nn.Module):
|
|
| 773 |
|
| 774 |
|
| 775 |
# output dir
|
| 776 |
-
save_path = 'resnet/model'
|
| 777 |
-
log_path = 'resnet/log'
|
| 778 |
|
| 779 |
if not os.path.exists(log_path):
|
| 780 |
os.mkdir(log_path)
|
|
|
|
| 773 |
|
| 774 |
|
| 775 |
# output dir
|
| 776 |
+
save_path = '/content/DDPM_ResNet_Unet/resnet/model'
|
| 777 |
+
log_path = '/content/DDPM_ResNet_Unet/resnet/log'
|
| 778 |
|
| 779 |
if not os.path.exists(log_path):
|
| 780 |
os.mkdir(log_path)
|