BVh}FdZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z dZd$dZ d%d Zd Zd Zed k(rej(Zej-dddej/deddej/deddej/deddej/ddddd dej/ddddd dej/dd ed!"ej3\ZZej6eej8dgez#yy)&z-A simple script for inspect checkpoint files.N)app) errors_impl)flags)py_checkpoint_readerc|j}|rotj|}i}d}d}|D]<}|j|r!|dz }|t j ||z }5||||<>|}t d|||fz|Dcgc]}t j ||}}t j|tScc}w)z Count total number of variables.rz?# Excluding %d tensors (%d params) that match %s when counting.)dtype) get_variable_to_shape_maprecompilesearchnpprodprintsumint) readercount_exclude_patternvar_to_shape_map regex_patternnew_var_to_shape_mapexclude_num_tensorsexclude_num_paramsv var_sizess Z/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/tools/inspect_checkpoint.py_count_total_paramsrs557JJ45M 6   a q bgg&6q&9::"21"5Q 6 , K/1FOH HI6FFrww'*+F)F   %%GsCFc f tj}|s|r|j}|j}t |j D]D\}} t d|d||jd| |s* t |j|Fn|s+t |jjddn~|j|st d|zy |j}|j}t d|d||jd||t |j|t d t|| zy #tj$rt dY.wxYw#t$r} t t!| d t!| vr t d dt!| vr]t#fddDrDdj%j'ddd} d} t | j)| Yd } ~ y Yd } ~ y Yd } ~ y d } ~ wwxYw)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 from count. ztensor: z (z) z"zutf-8ignore)errorsz!Tensor %s not found in checkpointNz# Total number of params: %drz#corrupted compressed block contentszFIt's likely that your checkpoint file has been compressed with SNAPPY.z Data lossc3&K|]}|v yw)N).0e file_names r z3print_tensors_in_checkpoint_file..js AqAN As)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 = {})rNewCheckpointReaderr get_variable_to_dtype_mapsorteditemsrname get_tensorr InternalError debug_stringdecode has_tensorr Exceptionstranyjoinsplitformat) r' tensor_name all_tensorsall_tensor_namesrrrvar_to_dtype_mapkeyvaluer& proposed_filev2_file_error_templates ` r print_tensors_in_checkpoint_filerC6s "): ! 5 5i @F&99;99;/55788*#u c+;C+@+E+EuMN  8 &##C( ) 8 F   ! ( ( ( BC   { + 1K?@99;99; *;7<< k * ,- F  k *+ (+>&;,= =>#** 8 6 7 8&  : #a&M,A6  s1v A$@ AAhhys3Ab9:m ' " ) )- 899 B  :sEA;FE"AF*A7F"FFFF H0B H++H0cD|jdd}t|dk7s|dstjd|z|\}}t j }||vrtjd|zt ||}|t durtjd|z |tur||n"tjj|}t jd i||iy#t$r$}tj|jd}~wwxYw) 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 numpy printoption. =r rz'%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$)r9lenargparseArgumentTypeErrorrget_printoptionstypeboolr BooleanParserparse ValueErrormessageset_printoptions)kv_strk_v_strkv_str printoptionsv_typerr&s rparse_numpy_printoptionrXss LLa '\Qgaj  $ $%Cf%L MM (!U$$&,l  $ $%G!%K LL  Q & tDz  $ $>B DD0   u #(#6#6#8#>#>u#E A 0  $ $QYY //0s'3C22 D;DDctjs!tdtjdyt tjtj tjtjtjy)NzUsage: inspect_checkpoint --file_name=checkpoint_file_name [--tensor_name=tensor_to_print] [--all_tensors] [--all_tensor_names] [--count_exclude_pattern] [--printoptions]r r") FLAGSr'rsysexitrCr;r<r=r) unused_argvs rmainr^sT   HHQK$ ** 511#99;__main__rKrLc(|jdk(S)Ntrue)lower)rs rrdsAGGI,?r_z --file_namez}Checkpoint filename. Note, if using Checkpoint V2 format, file_name is the shared prefix between all files in the checkpoint.)rKdefaulthelpz --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.)nargsconstrKrerfz--all_tensor_namesz,If True, print the names of all the tensors.z--printoptions*z9Argument for numpy.set_printoptions(), in the form 'k=v'.)rhrKrf)r^argv)r)Fr)__doc__rHr r[abslrnumpyrtensorflow.python.frameworkrtensorflow.python.platformrtensorflow.python.trainingrrZrrCrXr^__name__ArgumentParserparserregister add_argumentr6parse_known_argsunparsedrunrkr$r_rrzs~4 3,; &27<;=::z B;$ z "8 " " $&//&&"?@  ; <   * ,    N P      D F     9 ;   " F H ++-/% #''t388A;-(23Qr_