updated

generate_4ch.py

@@ -41,13 +41,13 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         description='generate images'
     )
-    parser.add_argument('--ckpt', type=str, default="/work/vajira/DL/FastGAN-pytorch/train_results/test1_4ch/models/all_50000.pth")
+    parser.add_argument('--ckpt', type=str, default="pre_trained_checkpoint_4ch/all_50000.pth")
     parser.add_argument('--artifacts', type=str, default=".", help='path to artifacts.')
     parser.add_argument('--cuda', type=int, default=0, help='index of gpu to use')
     parser.add_argument('--start_iter', type=int, default=6)
     parser.add_argument('--end_iter', type=int, default=10)
 
-    parser.add_argument('--dist', type=str, default='/work/vajira/DATA/FastGAN_polyps/test')
+    parser.add_argument('--dist', type=str, default='test_out')
     parser.add_argument('--size', type=int, default=256)
     parser.add_argument('--batch', default=1, type=int, help='batch size')
     parser.add_argument('--n_sample', type=int, default=1000)