support batch > 1 and add batch_size to model config. - Githubissues

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.

Apache License 2.0

310 stars 40 forks source link

support batch > 1 and add batch_size to model config. #56

Closed haozha111 closed 3 months ago

haozha111 commented 3 months ago

Remove the restriction of batch_size = 1. BUG=None