We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2c9c73d commit 9edefc3Copy full SHA for 9edefc3
1 file changed
src/maxdiffusion/trainers/wan_trainer.py
@@ -336,7 +336,6 @@ def training_loop(self, pipeline, optimizer, learning_rate_scheduler, train_data
336
if timestep not in eval_losses_by_timestep:
337
eval_losses_by_timestep[timestep] = []
338
eval_losses_by_timestep[timestep].append(l)
339
- print(f"timesteps: {timestep}, losses: {l}")
340
except StopIteration:
341
# This block is executed when the iterator has no more data
342
break
0 commit comments