o h @sddlZddlmZddlmZdgZdddZedd Zed d Zed d Z eddZ ddZ de e de e de e fddZdeeefdeddfddZdS)N)repeat)Any'consume_prefix_in_state_dict_if_presentparsecsfdd}||_|S)Ncs$t|tjjr t|Stt|SN) isinstance collectionsabcIterabletupler)xnj/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/nn/modules/utils.pyr sz_ntuple..parse)__name__)rnamerrr r_ntuple s r_single_pair_triple _quadruplecstfddt|DS)zReverse the order of `t` and repeat each element for `n` times. This can be used to translate padding arg used by Conv and Pooling modules to the ones used by `F.pad`. c3s"|] }tD]}|VqqdSr)range).0r _r rr s z(_reverse_repeat_tuple..)r reversed)trrr r_reverse_repeat_tuplesr"out_sizedefaultsreturncCsdddl}t|t|jfr|St|t|kr!tdt|dddt||t| dDS)Nrz#Input dimension should be at least rcSs g|] \}}|dur |n|qSrr)rvdrrr *sz&_list_with_default..)torchrintSymIntlen ValueErrorzip)r#r$r)rrr_list_with_default#sr/ state_dictprefixcCst|}|D]}||r|t|d}||||<qt|drUt|j}|D])}t|dkr6q-||ddksC||rT|t|d}|j||j|<q-dSdS)aStrip the prefix in state_dict in place, if any. .. note:: Given a `state_dict` from a DP/DDP model, a local model can load it by applying `consume_prefix_in_state_dict_if_present(state_dict, "module.")` before calling :meth:`torch.nn.Module.load_state_dict`. Args: state_dict (OrderedDict): a state-dict to be loaded to the model. prefix (str): prefix. N _metadatar.)listkeys startswithr,pophasattrr2replace)r0r1r6keynewkeyrrrr/s"    )r)r itertoolsrtypingr__all__rrrrrr"r5r*r/dictstrrrrrrs$      "