crfasrnn_keras icon indicating copy to clipboard operation
crfasrnn_keras copied to clipboard

Can you share your gpu verson?

Open Yc174 opened this issue 7 years ago • 2 comments

Thanks very much for your code,its really a good job! When I use your code to train my datasets on tensorflow/keras, it's really too slow.And I realize its reason that can only run on cpu! So can you give us your GPU verson? I will appreciate it! @sadeepj @sirotenko

Yc174 avatar Oct 24 '17 14:10 Yc174

For now, GPU implementation is only available for caffe https://github.com/hyenal/crfasrnn You can try to port it to TF op if you want

sirotenko avatar Oct 24 '17 14:10 sirotenko

@Yc174 Hi, is this repository able to train a new CRF-RNN model? I mean is backpro part included in "high_dim_filter.so" ? Thx/

XhqGlorry11 avatar Nov 17 '17 13:11 XhqGlorry11