def _InputParams(self):
    p = input_generator.NmtDoubleInput.Params()
    input_file = test_helper.test_src_dir_path(
        'tasks/mt/testdata/wmt14_ende_wpm_32k_doublebatch_test-000-001')
    p.file_pattern = 'tfrecord:' + input_file
    p.tokenizer.token_vocab_filepath = test_helper.test_src_dir_path(
        'tasks/mt/testdata/wmt14_ende_wpm_32k_test.vocab')
    p.file_random_seed = 31415
    p.file_parallelism = 1
    p.bucket_upper_bound = [10, 20]
    p.bucket_batch_limit = [4, 2]
    p.source_mask_ratio = -1
    p.source_mask_ratio_beta = '2,6'
    p.mask_word_id = 31999
    p.pad_id = 31998
    p.mask_words_ratio = 0.25
    p.permutation_distance = 3
    p.vocab_file = p.tokenizer.token_vocab_filepath
    p.packed_input = False
    return p