Skip to content

Commit b6fbbb7

Browse files
committed
fix bugs
1 parent 8fce15c commit b6fbbb7

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

users/zeineldeen/experiments/chunkwise_att_2023/tedlium2/configs/ted2_streaming.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -928,6 +928,11 @@ def train_mini_self_att(
928928

929929

930930
def compute_search_errors(exp_name, corpus_name, train_data, forward_args, model_checkpoint):
931+
returnn_root_fix = tk.Path(
932+
"/u/zeineldeen/setups/ubuntu_22_setups/2023-06-14--streaming-conf/returnn",
933+
hash_overwrite="returnn_root_fix_forward",
934+
)
935+
931936
returnn_forward_config = create_config(
932937
training_datasets=train_data,
933938
**forward_args,
@@ -938,7 +943,7 @@ def compute_search_errors(exp_name, corpus_name, train_data, forward_args, model
938943

939944
# add test dataset
940945
test_dataset = get_test_dataset_tuples(bpe_size=BPE_1K, selected_datasets=[corpus_name])
941-
returnn_forward_config.config["eval_datasets"] = {corpus_name: test_dataset[corpus_name][0].as_returnn_opts()}
946+
returnn_forward_config.config["forward_data"] = test_dataset[corpus_name][0].as_returnn_opts()
942947
# remove all others if exist
943948
returnn_forward_config.config.pop("train", None)
944949
returnn_forward_config.config.pop("dev", None)
@@ -968,7 +973,7 @@ def compute_search_errors(exp_name, corpus_name, train_data, forward_args, model
968973
model_checkpoint=model_checkpoint,
969974
returnn_config=config_,
970975
returnn_python_exe=RETURNN_CPU_EXE,
971-
returnn_root=RETURNN_ROOT,
976+
returnn_root=returnn_root_fix,
972977
hdf_outputs=["ground_truth_targets.hdf", "ground_truth_scores.hdf"],
973978
eval_mode=True,
974979
)
@@ -992,7 +997,7 @@ def compute_search_errors(exp_name, corpus_name, train_data, forward_args, model
992997
model_checkpoint=model_checkpoint,
993998
returnn_config=config_,
994999
returnn_python_exe=RETURNN_CPU_EXE,
995-
returnn_root=RETURNN_ROOT,
1000+
returnn_root=returnn_root_fix,
9961001
hdf_outputs=["search_output.hdf"],
9971002
eval_mode=False,
9981003
)

0 commit comments

Comments
 (0)