Train me. Punish me (offline)