def __init__(self, config, checkpoint_path, output_dir):
    super().__init__(config, checkpoint_path, output_dir)
def __init__(self, config, checkpoint_path, output_dir):
    super().__init__(config, checkpoint_path, output_dir)