missing newline

henryaddison · Sep 18, 2024 · 7f263e3 · 7f263e3
1 parent 90c40ce
commit 7f263e3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/ml_downscaling_emulator/score_sde_pytorch/run_lib.py b/src/ml_downscaling_emulator/score_sde_pytorch/run_lib.py
@@ -177,6 +177,7 @@ def train(config, workdir):
 
     if config.training.random_crop_size > 0:
       random_crop = torchvision.transforms.RandomCrop(config.training.random_crop_size)
+
     for epoch in range(initial_epoch, num_train_epochs + 1):
       state['epoch'] = epoch
       train_set_loss = 0.0