d9027c454d4765e83b52291bc1621bf4cc0508ee,luminoth/train_test.py,TrainTest,testTrainSave,#TrainTest#,127

Before Change

        // This is because of a MonitoredTrainingSession "bug".
        // When ending training it saves a checkpoint as the next step.
        // That causes that we are one step ahead when loading it.
        self.assertEqual(step, 5)


if __name__ == "__main__":
    tf.test.main()

After Change

        )

        // Get tensor from graph and run it in session
        w_tensor = tf.get_default_graph().get_tensor_by_name(
            "mockfasterrcnn/w:0"
        )
        w_numpy = new_session.run(w_tensor)

        // Assert we correctly loaded the weight
        self.assertArrayNear(w_numpy, [2.5, 3.0], err=0.01)

Instances