change to UserProducer in remote_trainer.py