def load_weights_from_given_checkpoint()

in activemri/experimental/cvpr19_models/trainer.py [0:0]

• 18 lines of code
• 6 McCabe index (conditional complexity)

    def load_weights_from_given_checkpoint(self):
        if self.options.weights_checkpoint is None:
            return
        elif not os.path.exists(self.options.weights_checkpoint):
            raise FileNotFoundError("Specified weights checkpoint do not exist!")
        self.logger.info(
            f"Loading weights from checkpoint found at {self.options.weights_checkpoint}."
        )
        checkpoint = torch.load(self.options.weights_checkpoint)
        self.reconstructor.load_state_dict(checkpoint["reconstructor"])
        if (
            self.options.use_evaluator
            and "evaluator" in checkpoint
            and checkpoint["evaluator"] is not None
        ):
            self.evaluator.load_state_dict(checkpoint["evaluator"])
        else:
            self.logger.info("Evaluator was not loaded.")