nit; "if the ``--train`` flag is used"