Input size mismatch

[pranjaliseth]

I'm trying to run the Transformer Relative model on my own dataset using the following command -

!onmt-main --model_type TransformerBaseRelative --config data.yml --auto_config train --with_eval

And my data.yml file is as following

data:
  train_features_file: korean_train.txt
  train_labels_file: english_train.txt
  eval_features_file: korean_val.txt
  eval_labels_file: korean_val.txt
  source_vocabulary: src-train.txt
  target_vocabulary: tgt-train.txt

eval:
  scorers: bleu
  early_stopping:
    metric: bleu
    min_improvement: 0.2
    steps: 4
train:
  max_step: 10000
  maximum_features_length: 70
  # (optional) The maximum length of label sequences during training (default: null).
  maximum_labels_length: 70
  

I am getting the below error -
'Input to reshape is a tensor with 8883180 values, but the requested shape has 22601160'

Kindly help me with the same.