diff --git a/models/multitask/esmm/config.yaml b/models/multitask/esmm/config.yaml index c6e79a52a6053c4441f253aca447c55307afa470..01c8fa61029e9eab1f4f594e0f1735df5737e6dd 100644 --- a/models/multitask/esmm/config.yaml +++ b/models/multitask/esmm/config.yaml @@ -24,7 +24,7 @@ dataset: - name: dataset_infer batch_size: 1 type: QueueDataset - data_path: "{workspace}/data/train" + data_path: "{workspace}/data/test" data_converter: "{workspace}/esmm_reader.py" hyper_parameters: