ffa271e34384438030753e46e20322ea8dc489a4,niftynet/engine/application_driver.py,ApplicationDriver,_randomly_init_or_restore_variables,#ApplicationDriver#Any#,202

Before Change

        // check model"s file
        checkpoint = "{}-{}".format(self.session_dir, self.initial_iter)
        assert tf.train.get_checkpoint_state(self.model_dir) is not None, \
            "Model file not found {}*, please check" \
            "config parameter: model_dir and *_iter".format(checkpoint)

        // restore session
        tf.logging.info("Accessing {} ...".format(checkpoint))
        self.saver.restore(sess, checkpoint)

After Change

            assert checkpoint, "checkpoint path not found " \
                               "in {}/checkpoints".format(self.model_dir)
            try:
                self.initial_iter = int(checkpoint.rsplit("-")[-1])
                tf.logging.info("set initial_iter to {} based "
                                "on checkpoints".format(self.initial_iter))
            except ValueError:
                tf.logging.fatal("failed to get interation number"
                                 "from checkpoint path")

Instances