Update resnet_wo_t/DDPM_ResNet_wo_t_sample.py

resnet_wo_t/DDPM_ResNet_wo_t_sample.py

@@ -832,10 +832,10 @@ save_each = 1
 
 diffusion_model = diffusion_model.to(device)
 
-last_trained_path = 'resnet_wo_t\model\epoch_30.pth'
-diffusion_model.load_state_dict(torch.load(os.path.join(last_trained_path))['model'])
+last_trained_path = '/content/DDPM_ResNet_Unet/resnet_wo_t/model/epoch_30.pth'
+diffusion_model.load_state_dict(torch.load(os.path.join(last_trained_path), map_location=device)['model'])
     
-sample_path = 'resnet_wo_t/sample2'
+sample_path = '/content/DDPM_ResNet_Unet/resnet_wo_t/sample'
 
 if not os.path.exists(sample_path):
     os.mkdir(sample_path)