hclhkbu / dlbench

Benchmarking State-of-the-Art Deep Learning Software Tools
http://dlbench.comp.hkbu.edu.hk/
MIT License
170 stars 47 forks source link

Tiny-dnn #8

Closed bhack closed 4 years ago

bhack commented 7 years ago

Are you interested in tiny-dnn with multiple engines?

shyhuai commented 7 years ago

It seems that tiny-dnn is outstanding on CPU platform, but it doesn't support GPU. Currently, we are mainly focusing on the frameworks that support GPU.

bhack commented 7 years ago

Yes it is on working. I will ping this issue when we will have also a full functional gpu code path.

shyhuai commented 7 years ago

That's great. Could you help to provide the scripts like current existing tools to running different networks (i.e., FCN-R, AlexNet-R, ResNet-56 and LSTM) after you are ready for GPU platform so that we can evaluate the performance in our hardware platform in time.