Update resnet/DDPM_ResNet_sample.py

resnet/DDPM_ResNet_sample.py

@@ -833,15 +833,15 @@ save_each = 1
 
 diffusion_model = diffusion_model.to(device)
 
-last_trained_path = 'resnet\model\epoch_30.pth'
+last_trained_path = '/content/DDPM_ResNet_Unet/resnet/model/epoch_30.pth'
 diffusion_model.load_state_dict(torch.load(os.path.join(last_trained_path))['model'])
     
-sample_path = 'resnet/sample2'
+sample_path = '/content/DDPM_ResNet_Unet/resnet/sample'
 
 if not os.path.exists(sample_path):
     os.mkdir(sample_path)
 
-num_sample = 500
+num_sample = 10000
 sample_batch = 16
 count = 0