Kiu-net-pytorch: bug: O tamanho do tensor a (3) deve corresponder ao tamanho do tensor b (6) na dimensão não-singleton 2

Você pode, por favor, adicionar mais descrição para este erro? Para qual conjunto de dados e qual modelo você está recebendo esse bug?

eu usei os dados LIST para treinar o modelo. obteve o .pth. então eu executo o val.py. (eu não alterei nenhum parâmetro em parameter.py, exceto esses caminhos). Então eu tenho o bug de models.py na linha 602. "O tamanho do tensor a (3) deve corresponder ao tamanho do tensor b (6) na dimensão 2 não-singleton". tentei alterar o "tamanho" do parameter.py para 8, não há bug, mas os resultados (pred.nii) parecem ruins. outro tamanho resultará em bug semelhante. eu também mudo o "tamanho" de paramter.py para 24 quando executo o train.py, porque também tenho bugs semelhantes.

sim, estou com o mesmo problema. o tamanho de saída (1024) não é do mesmo tamanho que o rótulo (512)

Tentando executar train.py na pasta LiTS após executar com sucesso get_training_set.py em data_prepare, mas recebi o seguinte erro ao usar kiunet_org como net em model.py:
Traceback (última chamada mais recente):
Arquivo "train.py", linha 71, em
saídas = net(ct)
Arquivo "/home/viplab/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", linha 727, em _call_impl
resultado = self.forward( entrada, * kwargs)
Arquivo "/home/viplab/anaconda3/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", linha 159, em frente
return self.module( entradas[0], * kwargs[0])
Arquivo "/home/viplab/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", linha 727, em _call_impl
resultado = self.forward( entrada, * kwargs)
Arquivo "/home/viplab/nas/KiU-Net-pytorch/LiTS/net/models.py", linha 591, em frente
out = torch.add(out,F.interpolate(F.relu(self.inte3_1bn(self.intere3_1(out1)))),scale_factor=(0.5,0.0625,0.0625),mode ='trilinear'))
RuntimeError: O tamanho do tensor a (3) deve corresponder ao tamanho do tensor b (6) na dimensão não-singleton 2

Eu tenho o mesmo problema ao executar (quase) qualquer um dos modelos em meus próprios dados:
out1 = torch.add(out1,F.interpolate(F.relu(self.inte1_2bn(self.intere1_2(tmp))),scale_factor=(4,4),mode ='bilinear')) RuntimeError: The size of tensor a (386) must match the size of tensor b (384) at non-singleton dimension 3

Eu tenho o mesmo problema, então como resolver?

Tem o mesmo problema kiunet_org.

estou com o mesmo problema alguem resolveu