the infer time is so long that I can't get a result.

[fan365]

`(cat) root@autodl-container-861840a478-c8fcbeca:~/project/CAT-Net# python tools/infer.py
<Splicing.data.dataset_arbitrary.arbitrary object at 0x7f454bcdbdd8>(15)
crop_size=None, grid_crop=True, blocks=('RGB', 'DCTvol', 'qtable'), mode=arbitrary, read_from_jpeg=True, class_weight=tensor([1., 1.])

---

output/splicing_dataset/CAT_full/CAT_full_v2.pth.tar
=> loading model from output/splicing_dataset/CAT_full/CAT_full_v2.pth.tar
Epoch: 196
0%| | 0/15 [00:00<?, ?it/s]`
I try to infer some image, but the infer time is so long that I wait for half an hour ,then I get no result

[fan365]

`+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 570.124.04 Driver Version: 570.124.04 CUDA Version: 12.8 |
|-----------------------------------------+------------------------+----------------------+
| GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|=========================================+========================+======================|
| 0 NVIDIA GeForce RTX 4090 On | 00000000:38:00.0 Off | Off |
| 30% 39C P2 68W / 450W | 2409MiB / 24564MiB | 0% Default |
| | | N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=========================================================================================|
| 0 N/A N/A 1735 C python 2400MiB |
+-----------------------------------------------------------------------------------------+` this is my env

[irTyone]

I meet the same probelm ,I only used author's example picture to run the infer.py, and output like this
:output/splicing_dataset/CAT_full/CAT_full_v2.pth.tar
=> loading model from output/splicing_dataset/CAT_full/CAT_full_v2.pth.tar
Epoch: 196
0%| | 0/1 [00:00<?, ?it/s]`