Update train_util.py

MaFengWoXiaoZi · Oct 24, 2022 · b7b6d89 · b7b6d89
1 parent 667a852
commit b7b6d89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/guided_diffusion/train_util.py b/guided_diffusion/train_util.py
@@ -185,7 +185,7 @@ def run_loop(self):
 
             self.run_step(batch, cond)
 
-            sample = self.run_step(batch, cond)
+
             i += 1
 
             if self.step % self.log_interval == 0: