added code for training benchmark on server