cbbb77338758d7cb20c5c29982f5aaac130f10d1,tests/application_driver_test.py,ApplicationDriverTest,test_multi_device_gradients,#ApplicationDriverTest#,148

Before Change

                        "worker_2/ComputeGradients/gradients/AddN_5:0"),
                    test_driver.graph.get_tensor_by_name(
                        "worker_3/ComputeGradients/gradients/AddN_5:0"),
                    test_driver.graph.get_tensor_by_name(
                        "ApplyGradients/Mean:0")
                ])
                msg = "same gradients for different devices"
                self.assertGreater(np.sum(np.abs(g_0 - g_1)), 0.0, msg)
                self.assertGreater(np.sum(np.abs(g_0 - g_2)), 0.0, msg)

After Change

            SESS_STARTED.send(test_driver.app, iter_msg=None)
            for i in range(2):
                sess.run(test_driver.app.gradient_op)
                g_0, g_1, g_2, g_3, g_ave = sess.run([
                    tf.get_default_graph().get_tensor_by_name(
                        "worker_0/ComputeGradients/gradients/AddN_5:0"),
                    tf.get_default_graph().get_tensor_by_name(
                        "worker_1/ComputeGradients/gradients/AddN_5:0"),
                    tf.get_default_graph().get_tensor_by_name(
                        "worker_2/ComputeGradients/gradients/AddN_5:0"),
                    tf.get_default_graph().get_tensor_by_name(
                        "worker_3/ComputeGradients/gradients/AddN_5:0"),
                    tf.get_default_graph().get_tensor_by_name(
                        "ApplyGradients/Mean:0")
                ])
                msg = "same gradients for different devices"
                self.assertGreater(np.sum(np.abs(g_0 - g_1)), 0.0, msg)
                self.assertGreater(np.sum(np.abs(g_0 - g_2)), 0.0, msg)

Instances