about test.py

[PatrickWilliams44]

Traceback (most recent call last): File "D:\CondaProjects\TransUNet-main\TransUNet\test.py", line 156, in inference(args, net, test_save_path) File "D:\CondaProjects\TransUNet-main\TransUNet\test.py", line 60, in inference metric_i = test_single_volume(image, label, model, classes=args.num_classes, File "D:\CondaProjects\TransUNet-main\TransUNet\utils.py", line 87, in test_single_volume out = torch.argmax(torch.softmax(net(input), dim=1), dim=1).squeeze(0) File "D:\Anaconda3\envs\python39\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl return forward_call(*input, **kwargs) File "D:\CondaProjects\TransUNet-main\TransUNet\networks\vit_seg_modeling.py", line 387, in forward x = x.repeat(1, 3, 1, 1) RuntimeError: Number of dimensions of repeat dims can not be smaller than number of dimensions of tensor

When I run test.py and the input channel of the testset is 3, how do I solve this problem?