Update resnet_wo_t/DDPM_ResNet_wo_t_sample.py
Browse files
resnet_wo_t/DDPM_ResNet_wo_t_sample.py
CHANGED
@@ -832,10 +832,10 @@ save_each = 1
|
|
832 |
|
833 |
diffusion_model = diffusion_model.to(device)
|
834 |
|
835 |
-
last_trained_path = 'resnet_wo_t
|
836 |
-
diffusion_model.load_state_dict(torch.load(os.path.join(last_trained_path))['model'])
|
837 |
|
838 |
-
sample_path = 'resnet_wo_t/
|
839 |
|
840 |
if not os.path.exists(sample_path):
|
841 |
os.mkdir(sample_path)
|
|
|
832 |
|
833 |
diffusion_model = diffusion_model.to(device)
|
834 |
|
835 |
+
last_trained_path = '/content/DDPM_ResNet_Unet/resnet_wo_t/model/epoch_30.pth'
|
836 |
+
diffusion_model.load_state_dict(torch.load(os.path.join(last_trained_path), map_location=device)['model'])
|
837 |
|
838 |
+
sample_path = '/content/DDPM_ResNet_Unet/resnet_wo_t/sample'
|
839 |
|
840 |
if not os.path.exists(sample_path):
|
841 |
os.mkdir(sample_path)
|