vlpr 6 months ago
parent 4377e655e2
commit 3121584af4

@ -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 = valid_steps(model=model, datasets=test_datasets, config=config, log_prefix=f"Model {config.model_name}")
results.to_csv(config.results_path) results.to_csv(config.results_path)
print(config.exp_dir.stem) print(config.exp_dir.stem, config.model_name)
print(results) print(results)
print() print()
print(f"Results saved to {config.results_path}") print(f"Results saved to {config.results_path}")

Loading…
Cancel
Save