@@ -928,6 +928,11 @@ def train_mini_self_att(
928928
929929
930930def 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