Skip to content
Snippets Groups Projects
Commit fc3551d9 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

Remove hardcoded predictions subfolder

parent be90dab0
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
......@@ -407,8 +407,7 @@ class PredictionsWriter(lightning.pytorch.callbacks.BasePredictionWriter):
for dataloader_idx, dataloader_name in enumerate(dataloader_names):
logfile = os.path.join(
self.output_dir,
f"predictions_{dataloader_name}",
"predictions.csv",
f"{dataloader_name}.csv",
)
os.makedirs(os.path.dirname(logfile), exist_ok=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment