ea1fbc2143f8c1bd5ff8f356e5611406b31feec2,tests/application_driver_test.py,ApplicationDriverTest,test_training_update,#ApplicationDriverTest#,97

Before Change

        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)
            train_op = test_driver.app.gradient_op

After Change

    def test_training_update(self):
        test_driver = get_initialised_driver()
        graph = test_driver.create_graph(test_driver.app, 1, 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)

Instances