jLddlmZddlmZddlZddlmZmZmZm Z  dddZ dS)) annotations)PathN)ARM64 IS_JETSONLOGGERYAMLmodeltorch.nn.Moduleim torch.Tensor output_dir Path | strmetadata dict | Noneprefixstrreturnc|tr Jdddlm}|tjrdn t rdnddfddl}dd lm }tj d |d |j d |||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.checksrtorchcuda is_availablerrx2paddle.convertrrinfo __version__rsaverr)r r rrrrrrs b/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/ultralytics/utils/export/paddle.py torch2paddler( s &CCCCC=;;;;;;z&&(( . , ,.&&-      OOO////// KUVUUHr.s#""""" <<<<<<<<<<<<! )))))))r)