search

hulinkuang / Cl-SegNet

Cl-SegNet
GNU General Public License v2.0
9 stars 1 forks source link

how could i use my own dataset? #5

Open VLAD-KONATA opened 4 months ago

VLAD-KONATA commented 4 months ago

well, i try to use my own dataset(after preprocessing with the nnUNet script，size=2，13，160，384) however,

Sizes of tensors must match except in dimension 1. Expected size 8 but got size 7 for tensor number 1 in the list. File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/network_architecture/cl_seg_aisd.py", line 433, in forward feat_trans = self.mfa2(torch.cat(feat_trans + [x_trnas[-1]], dim=1)) File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/training/network_training/nnUNetTrainerV2_AISD.py", line 246, in run_iteration output = self.network(data) File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/training/network_training/network_trainer.py", line 440, in run_training l = self.run_iteration(self.tr_gen, True) File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/training/network_training/nnUNetTrainer.py", line 315, in run_training super(nnUNetTrainer, self).run_training() File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/training/network_training/nnUNetTrainerV2_AISD.py", line 403, in run_training ret = super().run_training() File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/run/run_training.py", line 121, in main trainer.run_training() File "/home/konata/Git/Cl-SegNet/ClSeg_package/ClSeg/run/run_training.py", line 139, in main() RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 8 but got size 7 for tensor number 1 in the list.