Skip to content

GitLab

Commit 9d4fd3d3 authored by rprenger's avatar rprenger
Browse files

Fixing memory bug caused by DDP during task

parent 87023abd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tasks/zeroshot_gpt/evaluate.py
View file @ 9d4fd3d3
...@@ -205,7 +205,7 @@ def main(): ...@@ -205,7 +205,7 @@ def main():
args.task)) args.task))
# Set up model and load checkpoint. # Set up model and load checkpoint.
model = get_model(get_model_provider(eval_metric)) model = get_model(get_model_provider(eval_metric), wrap_with_ddp=False)
if args.load is not None: if args.load is not None:
_ = load_checkpoint(model, None, None) _ = load_checkpoint(model, None, None)
......
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