o ϯil@svddlmZmZmZmZddlmZddlmZdZ    dded e d e d e d eeeefd edej fddZ dS))AnyCallableDictOptionalN)load_state_dict_from_urlz5https://dl.fbaipublicfiles.com/pytorchvideo/model_zooFTmodel_builder_func pretrainedprogresscheckpoint_pathdefault_configkwargsreturnc Ks~|r t|dks Jd|dur$|D]\}}||dur#|||<q|di|}|r=t||dd} | d} || |S)a model_builder_func (Callable): Model builder function. pretrained (bool): Whether to load a pretrained model or not. Default: False. progress (bool): Whether or not to display a progress bar to stderr. Default: True. checkpoint_path (str): URL of the model weight to download. default_config (Dict): Default model configs that is passed to the model builder. **kwargs: (Any): Additional model configs. Do not modify the model configuration via the kwargs for pretrained model. rz*Do not change kwargs for pretrained model.Ncpu)r map_location model_state)lenitemsgetrload_state_dict) rr r r r r argumentvaluemodel checkpoint state_dictrrQ/home/ubuntu/.local/lib/python3.10/site-packages/pytorchvideo/models/hub/utils.pyhub_model_builder s r)FTrN)typingrrrrtorch.nnnn torch.hubrMODEL_ZOO_ROOT_DIRboolstrModulerrrrrs.