ea1fbc2143f8c1bd5ff8f356e5611406b31feec2,tests/application_driver_test.py,ApplicationDriverTest,test_multi_device_inputs,#ApplicationDriverTest#,116

Before Change

    def test_multi_device_inputs(self):
        test_driver = get_initialised_driver()
        test_driver.load_event_handlers(
            ["niftynet.engine.handler_sampler.SamplerThreading"])
        with self.test_session(graph=test_driver.create_graph()) as sess:
            sess.run(global_vars_init_or_restore())
            SESS_STARTED.send(test_driver.app, iter_msg=None)
            for i in range(2):
                sess.run(test_driver.app.gradient_op)
                s_0, s_1, s_2, s_3 = sess.run([

After Change

            test_driver.app.stop()

    def test_multi_device_inputs(self):
        test_driver = get_initialised_driver()
        graph = test_driver.create_graph(
            test_driver.app, test_driver.num_gpus, True)
        with self.test_session(graph=graph) as sess:
            SESS_STARTED.send(test_driver.app, iter_msg=None)
            GRAPH_FINALISING.send(test_driver.app, iter_msg=None)
            for i in range(2):
                sess.run(test_driver.app.gradient_op)
                s_0, s_1, s_2, s_3 = sess.run([
                    tf.get_default_graph().get_tensor_by_name(

Instances