(self, checkpoint)
Load parameters from checkpoint
Load parameters from checkpoint
Load parameters from checkpoint
def load_model(self, checkpoint):
        """Load parameters from checkpoint"""
        print(f'Load parameters from {checkpoint}')
        epoch = re.match(r"[0-9]*", os.path.basename(checkpoint)).group(0)
        self.epoch_i = int(epoch)
        self.model.load_state_dict(torch.load(checkpoint))
def load_model ( self , checkpoint ) : print ( f'Load parameters from {checkpoint}' ) epoch = re . match ( r"[0-9]*" , os . path . basename ( checkpoint ) ) . group ( 0 ) self . epoch_i = int ( epoch ) self . model . load_state_dict ( torch . load ( checkpoint ) )