fix bugs in sampling.py

ZERONE182 · May 23, 2023 · 1e80eed · 1e80eed
1 parent 87e4601
commit 1e80eed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffusion/train.py b/diffusion/train.py
@@ -66,7 +66,7 @@ def train(model, optimizer, loader, loader_val, writer, now, step, args):
         print(f'starting epoch {e}')
 
         for img, R, T, K, hue_delta in tqdm(loader):
-            validation(model, loader_val, writer, step, args.timesteps, args.batch_size)
+            # validation(model, loader_val, writer, step, args.timesteps, args.batch_size)
 
             warmup(optimizer, step, args.warmup_step / args.batch_size, args.lr)