train.py: Print elapsed time instead of time per epoch

bmdanielsson · bmdanielsson · commit 98e9e85de716 · 2022-11-22T19:25:31.000+01:00
diff --git a/train.py b/train.py
@@ -155,12 +155,12 @@ def main(args):
     scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='min', factor=0.1, patience=1, verbose=True, min_lr=1e-6)
 
     # Main training loop
+    start = time.monotonic()
     num_batches = len(train_data_loader)
     epoch = 0
     running_train_loss = 0.0
     while True:
         best_val_loss = 1000000.0
-        start = time.monotonic()
 
         for k, sample in enumerate(train_data_loader):
             train_loss = train_step(nnue, sample, optimizer, args.lambda_, epoch, k, num_batches)
