* fix build train_loop during test * fix build train_loop during test * fix build train_loop during test * fix build train_loop during test * Fix as comment