60dfcf812eca79017dced46e1189245c050a3fd6,utils/factory.py,UtilsFactory,unpack_checkpoint,#Any#Any#Any#Any#Any#,241

Before Change

        if optimizer is not None:
            for key in optimizer:
                optimizer[key].load_state_dict(
                    checkpoint["optimizer_" + str(key) + "_state_dict"])

        if scheduler is not None:
            for key in scheduler:
                scheduler[key] = checkpoint["scheduler_" + str(key)]

After Change

            else:
                model.load_state_dict(checkpoint["model_state_dict"])

        for dict2load, name2load in zip(
                [criterion, optimizer, scheduler],
                ["criterion", "optimizer", "scheduler"]):
            if dict2load is None:
                continue

            if isinstance(dict2load, dict):
                for key, value in dict2load.items():
                    if value is not None:
                        name2load_ = f"{name2load}_{key}_state_dict"
                        value.load_state_dict(checkpoint[name2load_])
            else:
                name2load = f"{name2load}_state_dict"
                dict2load.load_state_dict(checkpoint[name2load])

Instances