Tried to separate training from tagging. Failed