runtime error with inception_v3

aaron-xichen / pytorch-playground

Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)

MIT License

2.62k stars 612 forks source link

runtime error with inception_v3 #36

Closed minhson closed 4 years ago

minhson commented 4 years ago

when i run the command: python3 quantize.py --type inception_v3 --quant_method linear --param_bits 8 --fwd_bits 8 --bn_bits 10 --ngpu 4 --batch_size 64 --seed 1 i got the error as below: RuntimeError: The size of tensor a (3) must match the size of tensor b (864) at non-singleton dimension 3 library version: torch-1.3.1+cu92 torchvision-0.4.2+cu92

could you help me? thanks you!

aaron-xichen commented 4 years ago

fixed, please check