error

[Tan-yong]

Hi,Sorry to bother you.when I try to run tools/infer.py by args = argparse.Namespace(cfg='../experiments/CAT_DCT_only.yaml', opts=['TEST.MODEL_FILE', '../output/splicing_dataset/CAT_DCT_only/DCT_only_v2.pth.tar', 'TEST.FLIP_TEST', 'False', 'TEST.NUM_SAMPLES', '0']) and test_dataset = splicing_dataset(crop_size=None, grid_crop=True, blocks=('DCTvol', 'qtable'), DCT_channels=1, mode='arbitrary', read_from_jpeg=True) # DCT stream .

I just get: D:\conda\envs\cat\python.exe C:/Users/ty/Desktop/suanfayichengxu/CAT-Net-main/tools/infer.py <Splicing.data.dataset_arbitrary.arbitrary object at 0x00000206334F4F28>(0) crop_size=None, grid_crop=True, blocks=('DCTvol', 'qtable'), mode=arbitrary, read_from_jpeg=True, class_weight=tensor([1., 1.])

=> loading model from ../output/splicing_dataset/CAT_DCT_only/DCT_only_v2.pth.tar 0it [00:00, ?it/s]Epoch: 199 0it [00:01, ?it/s]

Process finished with exit code 0

[CauchyComplete]

Hi, It seems that the 'input' directory is empty. Put images there and run infer.py. Thanks

[Tan-yong]

thanks,it's work now,It's my GPU problem

[CauchyComplete]

Glad it works now!

[Cococc2]

谢谢，现在工作了，是我的GPU问题

I'm having the same problem. What can I do to fix it?