change path

xiaojize · Nov 5, 2021 · dfb1cca · dfb1cca
1 parent 3ea7822
commit dfb1cca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -465,7 +465,7 @@ def train(args, loader, generator, discriminator, g_optim, d_optim, g_ema, devic
     os.makedirs(f'checkpoint_{args.dataset}', exist_ok=True)
     os.makedirs(f'sample_{args.dataset}', exist_ok=True)
 
-    dataset = ImageFolder('../UI2I_via_StyleGAN2/cartoon', transform)
+    dataset = ImageFolder('./cartoon', transform)
     loader = data.DataLoader(
         dataset,
         batch_size=args.batch,