B лdQ@s>dZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z dZd)dd Zd*d d Zd dZddZedkr:eZeddddejdedddejdedddejdedddejddddd d d!ejd"dddd d#d!ejd$d%ed&d'e\ZZejeejdged(dS)+z-A simple script for inspect checkpoint files.N)app) errors_impl)flags)py_checkpoint_readercs||rxt|}i}d}d}x>D]6}||rR|d7}|t|7}q(|||<q(W|td|||ffddD}tj|tdS)z Count total number of variables.rz?# Excluding %d tensors (%d params) that match %s when counting.csg|]}t|qS)npprod).0v)var_to_shape_mapr\/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/tools/inspect_checkpoint.py 2sz'_count_total_params..)Zdtype) get_variable_to_shape_maprecompilesearchr r printsumint)readercount_exclude_patternZ regex_patternZnew_var_to_shape_mapZexclude_num_tensorsZexclude_num_paramsr Z var_sizesr)r r_count_total_paramss    rFc syt}|s|r|}|}xt|D]V\}} td|||j| f|r4yt||Wq4t j k rtdYq4Xq4Wnp|st| j dddnT| |std|dS|}|}td|||j||ft||tdt||d Wntk r} zvtt| d t| krLtd d t| krtfd ddDrdddd} d} t| | Wdd} ~ XYnXdS)a#Prints tensors in a checkpoint file. If no `tensor_name` is provided, prints the tensor names and shapes in the checkpoint file. If `tensor_name` is provided, prints the content of the tensor. Args: file_name: Name of the checkpoint file. tensor_name: Name of the tensor in the checkpoint file to print. all_tensors: Boolean indicating whether to print all tensors. all_tensor_names: Boolean indicating whether to print all tensor names. count_exclude_pattern: Regex string, pattern to exclude tensors when count. ztensor: %s (%s) %sz"zutf-8ignore)errorsz!Tensor %s not found in checkpointNz# Total number of params: %d)rz#corrupted compressed block contentszFIt's likely that your checkpoint file has been compressed with SNAPPY.z Data lossc3s|]}|kVqdS)Nr)r e) file_namerr jsz3print_tensors_in_checkpoint_file..)z.indexz.metaz.data.rz It's likely that this is a V2 checkpoint and you need to provide the filename *prefix*. Try removing the '.' and extension. Try: inspect checkpoint --file_name = {})rZNewCheckpointReaderrZget_variable_to_dtype_mapsorteditemsrnameZ get_tensorrZ InternalErrorZ debug_stringdecodeZ has_tensorr Exceptionstranyjoinsplitformat) r tensor_name all_tensorsall_tensor_namesrrr Zvar_to_dtype_mapkeyvaluerZ proposed_fileZv2_file_error_templater)rr print_tensors_in_checkpoint_file6sD     r0c Cs|dd}t|dks |ds.td||\}}t}||krTtd|t||}|tdkrztd|y"|tk r||n t |}Wn.t k r}zt|j Wdd}~XYnXtj f||idS) aSets a single numpy printoption from a string of the form 'x=y'. See documentation on numpy.set_printoptions() for details about what values x and y can take. x can be any option listed there other than 'formatter'. Args: kv_str: A string of the form 'x=y', such as 'threshold=100000' Raises: argparse.ArgumentTypeError: If the string couldn't be used to set any nump printoption. =rrz'%s' is not in the form k=v.z '%s' is not a valid printoption.Nz4Setting '%s' from the command line is not supported.)r)lenargparseArgumentTypeErrorr Zget_printoptionstypeboolrZ BooleanParserparse ValueErrormessageZset_printoptions)Zkv_strZk_v_strkZv_strZ printoptionsZv_typer rrrrparse_numpy_printoptionss    "r<cCs:tjstdtdnttjtjtjtjtj ddS)NzUsage: inspect_checkpoint --file_name=checkpoint_file_name [--tensor_name=tensor_to_print] [--all_tensors] [--all_tensor_names] [--printoptions]r)r) FLAGSrrsysexitr0r+r,r-r)Z unused_argvrrrmains r@__main__r6r7cCs |dkS)Ntrue)lower)r rrrrDz --file_namez}Checkpoint filename. Note, if using Checkpoint V2 format, file_name is the shared prefix between all files in the checkpoint.)r6defaulthelpz --tensor_namezName of the tensor to inspectz--count_exclude_patternzAPattern to exclude tensors, e.g., from optimizers, when counting.z --all_tensors?Tz7If True, print the names and values of all the tensors.)nargsconstr6rFrGz--all_tensor_namesz,If True, print the names of all the tensors.z--printoptions*z9Argument for numpy.set_printoptions(), in the form 'k=v'.)rIr6rG)r@argv)r)Fr)__doc__r4rr>Zabslrnumpyr Ztensorflow.python.frameworkrZtensorflow.python.platformrZtensorflow.python.trainingrr=rr0r<r@__name__ArgumentParserparserregister add_argumentr&parse_known_argsZunparsedrunrLrrrrsj      ;!