def __init__(self, args, src_dict, tgt_dict):
    super().__init__(args)
    self.src_dict = src_dict
    self.tgt_dict = tgt_dict
def __init__(self, args, src_dict, tgt_dict):
    super().__init__(args)
    self.src_dict = src_dict
    self.tgt_dict = tgt_dict