def __init__(self, checkpoint, prefix=None, map_location=None):
    self.checkpoint = checkpoint
    self.prefix = prefix
    self.map_location = map_location
def __init__(self, checkpoint, prefix=None, map_location=None):
    self.checkpoint = checkpoint
    self.prefix = prefix
    self.map_location = map_location