Open
Description
Hi @zijundeng and everybody,
First of all, thank you for your codes. I am a very beginner in semantic segmentation area.
I want to run your codes using SegNet model, but I didn't find the "train.py" for SegNet. Then, I simply try to modify "train.py" from folder "train/cityscapes-psp_net" and change the net
in line 42.
from : net = PSPNet(num_classes=cityscapes.num_classes)
to be : net = SegNet(num_classes=cityscapes.num_classes)
But, I found this error and still cannot solve the problem.
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 31 and 32 in dimension 2 at /opt/conda/conda-bld/pytorch_1525909934016/work/aten/src/THC/generic/THCTensorMath.cu:111
Please, kindly your suggestion.
Thank you.
Metadata
Metadata
Assignees
Labels
No labels