search

Jermmy / pytorch-quantization-demo

A simple network quantization demo using pytorch from scratch.
Apache License 2.0
497 stars 96 forks source link

Bugfix model device transfer #15

Closed genggng closed 2 years ago

genggng commented 2 years ago

修复了model在cpu和gpu设备转移时，部分成员变量无法正确随之转移，发生计算设备错误的bug。