def METHOD_NAME(self):
    # Train model for 1 iteration and do bmuf sync without doing warmup
    cfg, args = setup_args()
    iterations = 1
    results = self.bmuf_process(cfg, args, iterations)
    # Make sure params in both machines are same
    assert len(results) == 2
    self.assertAlmostEqual(results[0], results[1])