search

airockchip / rknn-toolkit2

Other
996 stars 104 forks source link

E RKNN: [14:52:58.948] Gru: Invalid batch_size of 40. Must be an integer equals 1 #152

Open myuzhao opened 2 months ago

myuzhao commented 2 months ago

模型转换时，当输入gru的batch size 大于1时报错

D RKNN: [14:52:58.942] >>>>>> start: rknn::RKNNAllocateConvCachePass D RKNN: [14:52:58.942] <<<<<<<< end: rknn::RKNNAllocateConvCachePass D RKNN: [14:52:58.942] >>>>>> start: OpEmit D RKNN: [14:52:58.943] finish initComputeZoneMapByStepsVector D RKNN: [14:52:58.943] finish initComputeZoneMapByStepsVector D RKNN: [14:52:58.943] finish initComputeZoneMapByStepsVector D RKNN: [14:52:58.943] finish initComputeZoneMapByStepsVector E RKNN: [14:52:58.948] Gru: Invalid batch_size of 40. Must be an integer equals 1

yuyun2000 commented 2 months ago

那就不要大于1呗