def _get_init_info(self):
    info = f'{self.__class__.__name__}: load from {self.checkpoint}'
    return info
def _get_init_info(self):
    info = f'{self.__class__.__name__}: load from {self.checkpoint}'
    return info