2aec65caac60fdc5ae6731b5142e231fc0614ed5,converter/nnef_converters/tf_converters/nnef_to_tf/nnef_to_tf.py,,convert,#Any#Any#Any#Any#Any#,163

Before Change

            nnef_dir_path = tmp_dir_name
            nnef_file_path = os.path.join(nnef_dir_path, "graph.nnef")
            export_variables = _has_dat_file(nnef_dir_path)
            if verbose and not export_variables:
                print("No weights were present in {}".format(nnef_path))
        elif nnef_path.endswith(".nnef"):
            nnef_dir_path = utils.without_file_name(nnef_path)
            nnef_file_path = nnef_path
            export_variables = False

After Change

        if verbose:
            print("Converting...")

        if os.path.isfile(nnef_path):
            assert nnef_path.endswith(".tgz") or nnef_path.endswith(".nnef"), \
                "Please specify a .nnef or a .tgz file or a directory"

        if os.path.isdir(nnef_path):
            nnef_dir_path = utils.without_slash(nnef_path)
            nnef_file_path = os.path.join(nnef_dir_path, "graph.nnef")
            export_variables = True

Instances