pytorch做resnet量化