Merge pull request #88 from frankwhzhang/master

fix esmm config

models/multitask/esmm/config.yaml

@@ -36,12 +36,11 @@ hyper_parameters:
     strategy: async
 
 #use infer_runner mode and modify 'phase' below if infer
-mode: train_runner
-#mode: infer_runner
+mode: [train_runner, infer_runner]
 
 runner:
 - name: train_runner
-  class: single_train
+  class: train
   device: cpu
   epochs: 3
   save_checkpoint_interval: 2
@@ -49,17 +48,20 @@ runner:
   save_checkpoint_path: "increment"
   save_inference_path: "inference"
   print_interval: 10
+  phases: [train]
 - name: infer_runner
-  class: single_infer
+  class: infer
   init_model_path: "increment/0"
   device: cpu
-  epochs: 3
+  epochs: 1
+  print_interval: 1
+  phases: [infer]
 
 phase:
-# - name: train
-#   model: "{workspace}/model.py"
-#   dataset_name: dataset_train
-#   thread_num: 1
+- name: train
+  model: "{workspace}/model.py"
+  dataset_name: dataset_train
+  thread_num: 1
 - name: infer
   model: "{workspace}/model.py"
   dataset_name: dataset_infer

models/multitask/esmm/data/test/small_test.txt

-0 16 
-475 473 155 
-491 21 
-96 185 96 
-29 14 13 
-5 481 11 21 470 
-70 5 70 11 
-167 42 167 217 
-72 15 73 161 172 
-82 82 
-97 297 97 
-193 182 186 183 184 177 214 
-152 152 
-163 298 7 
-39 73 71 
-490 23 23 496 488 74 23 74 486 23 23 74 
-17 17 
-170 170 483 444 443 234 
-25 472 
-5 5 11 70 69 
-149 149 455 
-356 68 477 468 17 479 66 
-159 172 6 71 6 6 158 13 494 169 
-155 44 438 144 500 
-156 9 9 
-146 146 
-173 10 10 461 
-7 6 6 
-269 48 268 
-50 100 
-323 174 18 
-69 69 22 98 
-38 171 
-22 29 489 10 
-0 0 
-11 5 
-29 13 14 232 231 451 289 452 229 
-260 11 156 
-166 160 166 39 
-223 134 134 420 
-66 401 68 132 17 84 287 5 
-39 304 
-65 84 132 
-400 211 
-145 144 
-16 28 254 48 50 100 42 154 262 133 17 
-0 0 
-28 28 
-11 476 464 
-61 61 86 86 
-38 38 
-463 478 
-437 265 
-22 39 485 171 98 
-434 51 344 
-16 16 
-67 67 67 448 
-22 12 161 
-15 377 147 147 374 
-119 317 0 
-38 484 
-403 499 
-432 442 
-28 0 16 50 465 42 
-163 487 7 162 
-99 99 325 423 83 83 
-154 133 
-5 37 492 235 160 279 
-10 10 457 493 10 460 
-441 4 4 4 4 4 4 4 
-153 153 
-159 164 164 
-328 37 
-65 65 404 347 431 459 
-80 80 44 44 
-61 446 
-162 495 7 453 
-157 21 204 68 37 66 469 145 
-37 151 230 206 240 205 264 87 409 87 288 270 280 329 157 296 454 474 
-430 445 433 
-449 14 
-9 9 9 9 
-440 238 226 
-148 148 
-266 267 181 
-48 498 
-263 255 256 
-458 158 7 
-72 168 12 165 71 73 173 49 
-0 0 
-7 7 6 
-14 29 13 6 15 14 15 13 
-480 439 21 
-450 21 151 
-12 12 49 14 13 165 12 169 72 15 15 
-91 91 
-22 12 49 168 
-497 101 30 411 30 482 30 53 30 101 176 415 53 447 
-462 150 150 
-471 456 131 435 131 467 436 412 227 218 190 466 429 213 326