092a0ec9e30a80fdec8f6be3b9b6b97c4e8ea472,examples/atari/reproduction/rainbow/train_rainbow.py,,main,#,26

Before Change

        if args.load:
            agent.load(args.load)
        else:
            agent.load(misc.download_model("Rainbow", args.env,
                                           model_type=args.pretrained_type))

    if args.demo:
        eval_stats = experiments.eval_performance(

After Change

        if args.load:
            agent.load(args.load)
        else:
            agent.load(misc.download_model("Rainbow", args.env,
                                           model_type=args.pretrained_type)[0])

    if args.demo:
        eval_stats = experiments.eval_performance(

Instances