PocketFlow
PocketFlow copied to clipboard
restore model at the end of `__train_pruned_model`
I want to know why restore model at the end of __train_pruned_model
, because no operation after this.
def __calc_grads_pruned(self, grads_origin):
......
if self.__is_primary_worker():
with self.pruner.model.g.as_default():
self.pruner.saver = tf.train.Saver()
# restore pruner model
self.pruner.saver.restore(self.pruner.model.sess, self.max_save_path)
Yes, this is redundant and can be removed. Thanks for pointing it out.