CenterNet_Pro_Max
CenterNet_Pro_Max copied to clipboard
Problem in training with resNet18 backbone
Thanks for the great job.
So far I m trying to train with "resnet18 + without DCN". But it occurs error as bellow.
File "/home/administrator/anaconda3/envs/CenterNet/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 778, in forward output_padding, self.groups, self.dilation) RuntimeError: Given transposed=1, weight of size 2048 256 4 4, expected input[8, 512, 16, 16] to have 2048 channels, but got 512 channels instead