OpenNMT-tf
OpenNMT-tf copied to clipboard
Input size mismatch
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.