def _flags_data(self, opdef, model_paths, local_cache):
main_mod = op_util.split_main(opdef.main)[0]
try:
flags_data = local_cache[main_mod]
except KeyError:
flags_data = self._flags_data_(main_mod, model_paths, opdef)
local_cache[main_mod] = flags_data
return flags_data
def python_script_opdef_loaded(self, opdef):
if opdef.output_scalars is not None:
return
assert opdef.main, opdef
main_mod = op_util.split_main(opdef.main)[0]
model_paths = op_util.opdef_model_paths(opdef)
try:
_path, mod_path = python_util.find_module(main_mod, model_paths)
except ImportError:
return
script = python_util.Script(mod_path)
if self.is_keras_script(script):
if opdef.output_scalars is None:
opdef.output_scalars = KERAS_OUTPUT_SCALARS
def python_script_opdef(self, op):
if (op.compare is not None and op.output_scalars is not None):
return
assert op.main, op
plugin = pluginlib.for_name("python_script")
main_mod = op_util.split_main(op.main)[0]
model_paths = op_util.opdef_model_paths(op)
try:
_sys_path, mod_path = plugin.find_module(main_mod, model_paths)
except ImportError:
return
script = python_util.Script(mod_path)
if self.is_keras_script(script):
if op.compare is None:
flags = ["=%s" % f.name for f in op.flags]
op.compare = flags + KERAS_BASE_COMPARE
if op.output_scalars is None:
op.output_scalars = KERAS_OUTPUT_SCALARS
