return self.fallback_
def save(self, finetune_obj, path, mkdir=True):
ckpt_reader = tf.train.load_checkpoint(finetune_obj.estimator_dir)
variable_map = ckpt_reader.get_variable_to_shape_map()
names = [name for name in variable_map.keys() if self.exclude_matches is None or self.exclude_matches not in name]
names = [name if name.endswith(":0") else name for name in names] // strip the :0 off the end
values = [ckpt_reader.get_tensor(name) for name in names]
After Change
return SaverHook()
def save(self, finetune_obj, path, mkdir=True):
names, values = self.variables.keys(), self.variables.values()
folder = os.path.dirname(path)
if not os.path.exists(folder) and mkdir:
os.mkdir(folder)