remove break from remote_trainer.py