search

xmartlabs / Bender

Easily craft fast Neural Networks on iOS! Use TensorFlow models. Metal under the hood.
https://xmartlabs.github.io/Bender/
MIT License
1.79k stars 90 forks source link

how to make bender data loader much fast? #74

Open erickingxu opened 7 years ago

erickingxu commented 7 years ago

I had met a problem, using blender for load some model's data from training framework,like pytorch. I need to convert them from float32 into float16(half) in Metal kernel for processing, but I found it is very slow, so is any way to speed up for this.

bryant1410 commented 7 years ago

I suggest you use benderthon to generate a protobuf and load it from bender. It's an easy and fast way to run a model.