aimet-model-zoo icon indicating copy to clipboard operation
aimet-model-zoo copied to clipboard

Published 20 hours ago • verified publisher icon quic

model.get_quantsim() load_checkpoint error

Open xianwujie opened this issue 4 months ago • 0 comments

Hi, I want to run the aimet_zoo_torch/mobilebert/evaluators/mobilebert_quanteval.py ,but I'm getting the error of load_checkpoint in model.get_quantsim(): ` File "/opt/conda/lib/python3.10/site-packages/aimet_torch/quantsim.py", line 2106, in load_checkpoint sim = pickle.load(file) TypeError: Enum.new() missing 1 required positional argument: 'value'

` my env:

  1. aimet-torch: 1.33.0
  2. torch: 2.2.2
  3. transformers: 4.38.1

Thanks for any reply.

xianwujie avatar Oct 21 '24 12:10 xianwujie