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

Fixing memory bug caused by DDP during task

parent 87023abd

tasks/zeroshot_gpt/evaluate.py

+1 −1
Original line number Diff line number Diff line
@@ -205,7 +205,7 @@ def main(): 
            args.task))



    # 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:

        _ = load_checkpoint(model, None, None)