Skip to content

Commit

Permalink
don't both recording val loss before any training
Browse files Browse the repository at this point in the history
it's large
  • Loading branch information
henryaddison committed Jul 8, 2024
1 parent 49dc1ef commit bf53f03
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions src/ml_downscaling_emulator/score_sde_pytorch/run_lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,12 +171,6 @@ def train(config, workdir):
if config.training.random_crop_size > 0:
random_crop = torchvision.transforms.RandomCrop(config.training.random_crop_size)

# log val loss before any training
if int(state['epoch']) == 0:
val_set_loss = val_loss(config, eval_dl, eval_step_fn, state)
epoch_metrics = {"epoch/val/loss": val_set_loss}
log_epoch(state['epoch'], epoch_metrics, wandb_run, writer)

for epoch in range(initial_epoch, num_train_epochs + 1):
state['epoch'] = epoch
train_set_loss = 0.0
Expand Down

0 comments on commit bf53f03

Please sign in to comment.