Update ball.py
Browse files
ball.py
CHANGED
|
@@ -382,7 +382,7 @@ def train_loop(config, model, noise_scheduler, optimizer, train_dataloader, lr_s
|
|
| 382 |
pipeline = DDPMPipeline(unet=accelerator.unwrap_model(model), scheduler=noise_scheduler)
|
| 383 |
|
| 384 |
if (epoch + 1) % config.save_image_epochs == 0 or epoch == config.num_epochs - 1:
|
| 385 |
-
evalfirst(config, epoch, pipeline)
|
| 386 |
evaluate(config, epoch, pipeline)
|
| 387 |
|
| 388 |
if (epoch + 1) % config.save_model_epochs == 0 or epoch == config.num_epochs - 1:
|
|
|
|
| 382 |
pipeline = DDPMPipeline(unet=accelerator.unwrap_model(model), scheduler=noise_scheduler)
|
| 383 |
|
| 384 |
if (epoch + 1) % config.save_image_epochs == 0 or epoch == config.num_epochs - 1:
|
| 385 |
+
#evalfirst(config, epoch, pipeline)
|
| 386 |
evaluate(config, epoch, pipeline)
|
| 387 |
|
| 388 |
if (epoch + 1) % config.save_model_epochs == 0 or epoch == config.num_epochs - 1:
|