search

MASILab / 3DUX-Net

244 stars 34 forks source link

RuntimeError: Given groups=1, weight of size unmatch #25

Closed chunniunai220ml closed 1 year ago

chunniunai220ml commented 1 year ago

image

hi, the test_seg.py ,--dataset flare --network 3DUXNET ,i have tested last week ,is ok, but now it occurs weights unmatch, i see the hidden_size=768

leeh43 commented 1 year ago

Have you modified the network design? If yes, do you mind to guide me that where have you changed?

chunniunai220ml commented 1 year ago

nope, i just git clone the repo, and follow the test insturction, python test_seg.py --root ./labelsTs-public30/ --output res --dataset flare --network 3DUXNET --trained_weights ./models/best_metric_model.pth --mode test --sw_batch_size 4 --overlap 0.7 --gpu 0 --cache_rate 0.2 i tested suceessfully last week,but it's wired that i met this bug, i see the network architecture is right

leeh43 commented 1 year ago

Seems like your command is using labelsTs for input, instead of images?