search

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

does it work for conv2d_transpose? #28

Closed bailin7134 closed 4 years ago

bailin7134 commented 5 years ago

does it work for conv2d_transpose? I notice in quant.py, it works for nn.Conv2d, nn.Linear, nn.BatchNorm1d, nn.BatchNorm2d, nn.AvgPool2d. If I add nn.Conv2d_transpose, will it work?

aaron-xichen commented 4 years ago

It will add quantization layer after all nn.Conv2d_transpose layers