def load_self()

in model/interpolation_net.py [0:0]

• 17 lines of code
• 3 McCabe index (conditional complexity)

    def load_self(self, folder_path, num_epoch=None):
        if num_epoch is None:
            ckpt_name = "ckpt_last.pth"
            ckpt_path = os.path.join(folder_path, ckpt_name)
        else:
            ckpt_name = "ckpt_ep{}.pth".format(num_epoch)
            ckpt_path = os.path.join(folder_path, ckpt_name)

        self.load_chkpt(ckpt_path)

        if num_epoch is None:
            print("Loaded model from ", folder_path, " with the latest weights")
        else:
            print(
                "Loaded model from ",
                folder_path,
                " with the weights from epoch ",
                num_epoch,
            )