search

inisis / brocolli

Everything in Torch Fx
MIT License
341 stars 63 forks source link

test_quanzation_nets 报错解决 #58

Closed xibici closed 1 year ago

xibici commented 1 year ago

1.win10/11情况下,需要删除dataloader的 num_workers=8 否则会出以下错 E AttributeError: Can't pickle local object 'test_mnist..MNISTCHINA'

2.#121 pytorch_quantizer.prepare()改成 pytorch_quantizer.prepare_calibration()