Update reference to batch accumulation flag. #69

src/mednet/scripts/upload.py

@@ -187,9 +187,9 @@ def upload(
             mlflow.log_param(k, train_data[k])
 
         click.secho(
-            f"  -> `#accumulations` ({train_data['batch-chunk-count']})"
+            f"  -> `#accumulations` ({train_data['accumulate-grad-batches']})"
         )
-        mlflow.log_param("#Accumulations", train_data["batch-chunk-count"])
+        mlflow.log_param("#Accumulations", train_data["accumulate-grad-batches"])
         click.secho(f"  -> `epoch (best)` ({best_epoch})")
         mlflow.log_param("Epoch (best)", best_epoch)