How to export the SwinT model from BEVFusion?

[slayerlpj]

I have trained camera+lidar based on SwinT, I have got a pth. How can I generate PTQ model or FP16 model?

I used the command "python qat/ptq.py --config=configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/configs.yaml --ckpt=model/swint/epoch_30.pth --calibrate_batch 300", the I got a error as follow:

2024-08-02 09:16:14,038 - mmdet - INFO - load checkpoint from http path: https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_tiny_patch4_window7_224.pth I0802 09:16:14.038485 139857851803456 logging.py:101] load checkpoint from http path: https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_tiny_patch4_window7_224.pth load checkpoint from local path: model/swint/epoch_30.pth Traceback (most recent call last): File "qat/ptq.py", line 150, in main() File "qat/ptq.py", line 130, in main model = quantize_net(model) File "qat/ptq.py", line 81, in quantize_net quantize.quantize_encoders_camera_branch(model.encoders.camera) File "/home/work/work/lpj/BEVFusion/Lidar_AI_Solution/CUDA-BEVFusion/qat/lean/quantize.py", line 175, in quantize_encoders_camera_branch major = model_camera_branch.backbone.layer3[0].conv1._input_quantizer File "/home/work/anaconda3/envs/bevfusion/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1177, in getattr raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'SwinTransformer' object has no attribute 'layer3'

How can I fix the problem?