NVlabs / pacnet

Pixel-Adaptive Convolutional Neural Networks (CVPR '19)
https://suhangpro.github.io/pac/
Other
515 stars 79 forks source link

About how to train a fcn8spac from scratch, thanks #30

Open fullsky-1995 opened 2 years ago

fullsky-1995 commented 2 years ago

Hi, I want to use the pacconv2d in my code, so I try to running your code firstly but I meet a proplem. The PacConv2dFn in pac.py: in_mul_k = cols.view(bs, ch, *kernel.shape[2:]) * kernel The error is: RuntimeError: shape '[1, 256, 3, 3, 112, 112]' is invalid for input of size 29419776 I train the model with the command as: >python -m task_semanticSegmentation.main --data-root data/voc --exp-root exp/voc/fcn8spac2 --train-split train11 --test-split val11_sbd --train-crop 449 --test-crop 512 --model fcn8spac --epochs 40 --lr 0.001 --lr-steps 20 How can I solve this problem or whether do my parameter exit error? My environment is: python 3.7, cuda 11.1 and pytorch 1.9