diff --git a/src/validate.py b/src/validate.py index ace279a..f2d6d2f 100644 --- a/src/validate.py +++ b/src/validate.py @@ -87,7 +87,7 @@ if __name__ == "__main__": results = valid_steps(model=model, datasets=test_datasets, config=config, log_prefix=f"Model {config.model_name}") results.to_csv(config.results_path) - print(config.exp_dir.stem) + print(config.exp_dir.stem, config.model_name) print(results) print() print(f"Results saved to {config.results_path}")