Update unet_wo_t/DDPM_Unet_wo_t_sample.py

unet_wo_t/DDPM_Unet_wo_t_sample.py

@@ -909,17 +909,17 @@ save_each = 1
 
 diffusion_model = diffusion_model.to(device)
 
-last_trained_path = 'unet_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/unet_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 = 'unet_wo_t/sample'
+sample_path = '/content/DDPM_ResNet_Unet/unet_wo_t/sample'
 
 if not os.path.exists(sample_path):
     os.mkdir(sample_path)
 
-num_sample = 900
+num_sample = 10000
 sample_batch = 16
-count = 9168
+count = 0
 
 if num_sample % sample_batch != 0:
     num_sample = num_sample + (sample_batch - (num_sample % sample_batch))