Commit 3654b7f3 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix inference script again

parent c22d0111
...@@ -342,7 +342,7 @@ def main(args): ...@@ -342,7 +342,7 @@ def main(args):
) )
for model, model_version in load_models_from_command_line(args, config): for model, model_version in load_models_from_command_line(args, config):
working_batch = deepcopy(batch) working_batch = deepcopy(processed_feature_dict)
out = run_model(model, working_batch, tag, args) out = run_model(model, working_batch, tag, args)
# Toss out the recycling dimensions --- we don't need them anymore # Toss out the recycling dimensions --- we don't need them anymore
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment