o h @sdZddlZddlmZddlmZddlmZm Z GdddeZ   dd ej j d ee ed eee d ed ej jf ddZd ej j fddZd ej j d eed eefddZdS)zKThis module contains utility method for mobile model optimization and lint.N)Enum)_MobileOptimizerType)OptionalAnyStrc@seZdZdZdZdZdZdS)LintCodeN)__name__ __module__ __qualname__ BUNDLED_INPUT REQUIRES_GRADDROPOUT BATCHNORMrrp/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/utils/mobile_optimizer.pyr s rCPU script_moduleoptimization_blocklistpreserved_methodsbackendreturncsttjjstdtd|durt}|durg}dd|D}t|}tfdd|Dr=t t||}fdd|D}|rRt d d || }|d krdtj j||}n|d krrtj j||}n|d krtj j|}ntdtjj|S)a Optimize a torch script module for mobile deployment. Args: script_module: An instance of torch script module with type of ScriptModule. optimization_blocklist: A set with type of MobileOptimizerType. When set is not passed, optimization method will run all the optimizer pass; otherwise, optimizer method will run the optimization pass that is not included inside optimization_blocklist. preserved_methods: A list of methods that needed to be preserved when freeze_module pass is invoked backend: Device type to use for running the result model ('CPU'(default), 'Vulkan' or 'Metal'). Returns: A new optimized torch script module Got , but ScriptModule is expected.NcSsg|]}t|qSr)str.0methodrrr .sz'optimize_for_mobile..c3s|]}t|VqdS)Nhasattrrrrr 1sz&optimize_for_mobile..csg|] }t|s|qSrr!rr#rrr 4szAThe following methods to preserve do not exist in script_module: z, cpuvulkanmetalz:Unknown backend, must be one of 'CPU', 'Vulkan' or 'Metal') isinstancetorchjit ScriptModule TypeErrortypeset(_get_bundled_inputs_preserved_attributesalllistAttributeErrorjoinlower_C_jit_pass_optimize_for_mobile_c$_jit_pass_vulkan_optimize_for_mobile#_jit_pass_metal_optimize_for_mobile _recursivewrap_cpp_module)rrrrpreserved_methods_strbundled_inputs_attributesnon_exist_methodsoptimized_cpp_modulerr#roptimize_for_mobilesD r@cCst|tjjstdt|dg}t|ds"|tj j dd| D]\}}|j r;|tj j d|ddq&tj|}|D]&}d|vrX|tjj d |d dd |vrj|tjj d |d dqD|S) z Generate a list of lints for a given torch script module. Args: script_module: An instance of torch script module with type of ScriptModule. Returns: lint_map: A list of dictionary that contains modules lints rr$_generate_bundled_inputs_for_forwardzNo bundled input for forward, please add bundled inputs before saving the module using torch.utils.bundled_inputs.augment_model_with_bundled_inputs.)namemessagezParam zw requires grad, please set torch.no_grad() to reduce memory usage and improve computation speed during inference phase.dropoutz Operator z exists, remember to call eval() before saving the module.and call torch.utils.mobile_optimizer.optimize_for_mobile to drop dropout operator. batch_normz exists, remember to call eval() before saving the module and call torch.utils.mobile_optimizer.optimize_for_mobile to drop batch_norm operator.)r(r)r*r+r,r-r"appendrrrBnamed_parameters requires_gradrexport_opnamesrr)r lint_listrBparamop_namesop_namerrrgenerate_mobile_module_lintsLs.       rNcCszg}t|dr|d|dt|dr;|d|}|D]}||vr,|||d||d|q!|S)Nget_all_bundled_inputsget_num_bundled_inputs%get_bundled_inputs_functions_and_infoget_all_bundled_inputs_for__bundled_inputs_deflated_)r"rFrQ)rrr=all_info function_namerrrr/us      r/)NNr)__doc__r)enumrtorch._CrMobileOptimizerTypetypingrrrr*r+r.r1rRecursiveScriptModuler@rNr/rrrrs,     =&)