76b4d1ede08ef3c9b59106269e088fbce3a3689a,dl/callbacks.py,CheckpointCallback,load_checkpoint,#,277

Before Change

                if isinstance(model, torch.nn.DataParallel):
                    model = model.module
                if isinstance(model, Fp16Wrap):
                    model.network.load_state_dict(
                        checkpoint["model_state_dict"])
                else:
                    model.load_state_dict(
                        checkpoint["model_state_dict"])

After Change

            state.epoch = checkpoint["epoch"]
            state.best_metrics = checkpoint["best_metrics"]

            UtilsFactory.unpack_checkpoint(
                checkpoint,
                model=state.model, criterion=state._criterion,
                optimizer=state._optimizer, scheduler=state._scheduler)

            print("loaded checkpoint \"{}\" (epoch {})"
                  .format(filename, checkpoint["epoch"]))
        else:

Instances