fix dselectK

Author: wangzhen38

models/multitask/dselect_k/config.yaml

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 runner:
-  train_data_dir: "data/sample_data"
+  train_data_dir: "data/sample_data.pickle"
   train_reader_path: "multiMNIST_reader" # importlib format
   use_gpu: False
   use_auc: True
@@ -22,7 +22,7 @@ runner:
   print_interval: 2
   #model_init_path: "output_model/0" # init model
   model_save_path: "output_model_dselect_k"
-  test_data_dir: "data/sample_data"
+  test_data_dir: "data/sample_data.pickle"
   infer_batch_size: 8
   infer_reader_path: "multiMNIST_reader" # importlib format
   infer_load_path: "output_model_dselect_k"

models/multitask/dselect_k/config_bigdata.yaml

@@ -23,7 +23,7 @@ runner:
   #model_init_path: "output_model/0" # init model
   model_save_path: "output_model_dselect_k_all"
   test_data_dir: "../../../datasets/Multi_MNIST_DselectK/test"
-  infer_batch_size: 20000
+  infer_batch_size: 256
   infer_reader_path: "multiMNIST_reader" # importlib format
   infer_load_path: "output_model_dselect_k_all"
   infer_start_epoch: 0

models/multitask/dselect_k/readme.md

@@ -68,8 +68,7 @@
 
 上图是其中一张图片，与经典数据集 MNIST 不同，该图片上包含两位数字，左上和右下，对应两个多分类任务。数据集划分为训练集、验证集、测试集，数量分别为 100000、20000、20000。
 
-在 PaperswithCode 网站上检索到 【NeurlPS 2019】[Pareto Multi-Task Learning](https://arxiv.org/pdf/1912.12854v1.pdf) 公布了该数据集，
-下载链接： https://drive.google.com/drive/folders/1VnmCmBAVh8f_BKJg1KYx-E137gBLXbGG。
+在 PaperswithCode 网站上检索到 【NeurlPS 2019】[Pareto Multi-Task Learning](https://arxiv.org/pdf/1912.12854v1.pdf) 公布了该数据集。
 
 
 ## 运行环境