diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py index 1bd1add1c98b0bdc56e57dc559f073ff5c8ce030..e3a4ff75c86baa2ad34b4ccd8527e95ca8a192da 100644 --- a/src/mednet/scripts/upload.py +++ b/src/mednet/scripts/upload.py @@ -157,7 +157,7 @@ def upload( evaluation_data = evaluation_data["test"] # get lowest validation epoch - best_epoch = str(train_model_file).split('.')[0].split('=')[1] + best_epoch = str(train_model_file).split(".")[0].split("=")[1] experiment_name = ( experiment_name @@ -185,9 +185,11 @@ def upload( ]: click.secho(f" -> `{k}` ({train_data[k]})") mlflow.log_param(k, train_data[k]) - - click.secho(f" -> `#accumulations` ({train_data['batch-chunk-count']})") - mlflow.log_param("#Accumulations", train_data['batch-chunk-count']) + + click.secho( + f" -> `#accumulations` ({train_data['batch-chunk-count']})" + ) + mlflow.log_param("#Accumulations", train_data["batch-chunk-count"]) click.secho(f" -> `epoch (best)` ({best_epoch})") mlflow.log_param("Epoch (best)", best_epoch)