Skip to content
Snippets Groups Projects
Commit d076a176 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[scripts.experiment] Resync with changes to other scripts

parent 01ae2cbb
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
Pipeline #77154 passed
...@@ -300,15 +300,17 @@ def experiment( ...@@ -300,15 +300,17 @@ def experiment(
if not os.path.exists(model_file): if not os.path.exists(model_file):
model_file = os.path.join(train_output_folder, "model_final_epoch.ckpt") model_file = os.path.join(train_output_folder, "model_final_epoch.ckpt")
predictions_folder = os.path.join(output_folder, "predictions") predictions_output = os.path.join(output_folder, "predictions.json")
ctx.invoke( ctx.invoke(
predict, predict,
output_folder=predictions_folder, output=predictions_output,
model=model, model=model,
datamodule=datamodule, datamodule=datamodule,
device=device, device=device,
weight=model_file, weight=model_file,
batch_size=batch_size,
parallel=parallel,
) )
logger.info("Ended predicting") logger.info("Ended predicting")
...@@ -322,7 +324,7 @@ def experiment( ...@@ -322,7 +324,7 @@ def experiment(
ctx.invoke( ctx.invoke(
evaluate, evaluate,
output_folder=evaluations_folder, output_folder=evaluations_folder,
predictions=os.path.join(predictions_folder, "predictions.json"), predictions=predictions_output,
threshold="validation", threshold="validation",
) )
......
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