Fix task_parallel_based/train.cpp