Tencent / PocketFlow

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.
https://pocketflow.github.io
Other
2.79k stars 490 forks source link

mobilenetv2 faster rcnn uniform 量化训练报错 #300

Open zyxcambridge opened 5 years ago

zyxcambridge commented 5 years ago

File "/opt/project/learners/uniform_quantization/learner.py", line 92, in init self.build_train() # for train File "/opt/project/learners/uniform_quantization/learner.py", line 193, in build_train images.set_shape((FLAGS.batch_size, images.shape[1], images.shape[2], AttributeError: 'dict' object has no attribute 'set_shape'

怎么修改尼