special_tokens = {
  '<pad>': 0,
  '<sos>': 1,
  '<eos>': 2,
}

PAD_ID = special_tokens["<pad>"]
SOS_ID = special_tokens["<sos>"]
EOS_ID = special_tokens["<eos>"]