Open Yffy123456 opened 1 year ago
/home/wf/anaconda3/envs/nnFormer0/bin/python /home/wf/nnFormerFrame/nnFormer0/nnformer/network_architecture/nnFormer_acdc.py
Traceback (most recent call last):
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/network_architecture/nnFormer_acdc.py", line 971, in
raceback (most recent call last): File "/home/wf/nnFormerFrame/nnFormer0/nnformer/inference/predict_simple.py", line 202, in
main()
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/inference/predict_simple.py", line 200, in main
step_size=step_size, checkpoint_name=args.chk)
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/inference/predict.py", line 637, in predict_from_folder
segmentation_export_kwargs=segmentation_export_kwargs)
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/inference/predict.py", line 221, in predict_cases
mixed_precision=mixed_precision)[1][None])
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/training/network_training/nnFormerTrainerV2_nnformer_tumor.py", line 253, in predict_preprocessed_data_return_seg_and_softmax
mixed_precision=mixed_precision)
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/training/network_training/nnFormerTrainer.py", line 526, in predict_preprocessed_data_return_seg_and_softmax
mixed_precision=mixed_precision)
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/network_architecture/neural_network.py", line 150, in predict_3D
verbose=verbose)
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/network_architecture/neural_network.py", line 387, in _internal_predict_3D_3Dconv_tiled
gaussian_importance_map)[0]
File "/home/wf/nnFormerFrame/nnFormer0/nnformer/network_architecture/neural_network.py", line 532, in _internal_maybe_mirror_and_pred_3D
result_torch += 1 / num_results * pred
RuntimeError: The size of tensor a (128) must match the size of tensor b (64) at non-singleton dimension 2
When i in predict have problem 529 print shape print('529',num_results,pred.shape,result_torch.shape)
529 8 torch.Size([1, 4, 64, 128, 128]) torch.Size([1, 4, 128, 128, 128])