Fi`ddlmZddlmZddlZddlmZmZmZm Z  d ddZ y)) annotations)PathN)ARM64 IS_JETSONLOGGERYAMLcdtrJdddlm}|tjj rdn t rdnddfddl}dd lm }tjd |d |jd |||d |g|r"tjt|dz |t!|S)aExport a PyTorch model to PaddlePaddle format using X2Paddle. Args: model (torch.nn.Module): The PyTorch model to export. im (torch.Tensor): Example input tensor for tracing. output_dir (Path | str): Directory to save the exported PaddlePaddle model. metadata (dict | None): Optional metadata saved as ``metadata.yaml``. prefix (str): Prefix for log messages. Returns: (str): Path to the exported ``_paddle_model`` directory. z'Jetson Paddle exports not supported yetr)check_requirementszpaddlepaddle-gpu>=3.0.0,<3.3.0zpaddlepaddle==3.0.0zpaddlepaddle>=3.0.0,<3.3.0x2paddleN)pytorch2paddle z starting export with X2Paddle z...trace)modulesave_dirjit_typeinput_examplesz metadata.yaml)rultralytics.utils.checksr torchcuda is_availablerr x2paddle.convertr rinfo __version__rsaverstr)modelim output_dirmetadataprefixr r r s a/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/utils/export/paddle.py torch2paddler" s&CCC=;zz&&( -'-    / KK"VH;Hr*s[# <<! ) ))) )  )  )r#