DwangoMediaVillage / keras_compressor

Model Compression CLI Tool for Keras.
https://nico-opendata.jp/ja/casestudy/model_compression/index.html
157 stars 39 forks source link

how can run it on gpu? #16

Open taheritajar opened 5 years ago

taheritajar commented 5 years ago

I use compress.py, but it is very slow with my model. how can run it on gpu?