Kiu-net-pytorch: bug : la taille du tenseur a (3) doit correspondre à la taille du tenseur b (6) à la dimension non singleton 2

Pouvez-vous s'il vous plaît ajouter plus de description pour cette erreur? Pour quel jeu de données et quel modèle avez-vous ce bug ?

J'ai utilisé les données LIST pour former le modèle. obtenu le .pth. puis j'exécute le val.py. (je n'ai modifié aucun paramètre dans parameter.py, à l'exception de ces chemins). Ensuite, j'ai eu le bogue de models.py à la ligne 602. "La taille du tenseur a (3) doit correspondre à la taille du tenseur b (6) à la dimension non singleton 2". j'ai essayé de changer la "taille" de parameter.py à 8, il n'y a pas de bogue, mais les résultats (pred.nii) semblent mauvais. une autre taille entraînera un bogue similaire. Je change également la "taille" de paramter.py à 24 lorsque je lance le train.py, car j'ai aussi des bogues similaires.

oui, j'ai le même problème. la taille de sortie (1024) n'est pas la même que celle de l'étiquette (512)

Essayer d'exécuter train.py dans le dossier LiTS après avoir exécuté avec succès get_training_set.py dans data_prepare, mais a obtenu l'erreur suivante lors de l'utilisation de kiunet_org en tant que net dans model.py :
Traceback (dernier appel le plus récent) :
Fichier "train.py", ligne 71, dans
sorties = net(ct)
Fichier "/home/viplab/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", ligne 727, dans _call_impl
result = self.forward( input, * kwargs)
Fichier "/home/viplab/anaconda3/lib/python3.8/site-packages/torch/nn/parallel/data_parallel.py", ligne 159, en avant
return self.module( entrées[0], * kwargs[0])
Fichier "/home/viplab/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", ligne 727, dans _call_impl
result = self.forward( input, * kwargs)
Fichier "/home/viplab/nas/KiU-Net-pytorch/LiTS/net/models.py", ligne 591, en avant
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'))
Erreur d'exécution : la taille du tenseur a (3) doit correspondre à la taille du tenseur b (6) à la dimension non singleton 2

J'ai le même problème avec l'exécution (presque) de l'un des modèles sur mes propres données :
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

J'ai le même problème，Alors comment le résoudre?

Avoir le même problème kiunet_org.

j'ai le même problème quelqu'un l'a t'il résolu