neupy icon indicating copy to clipboard operation
neupy copied to clipboard

Will there be GPU support for competitive models?

Open lejinghu opened this issue 6 years ago • 5 comments

I know this is a probably a difficult task. But since we are switching to Tensorflow, maybe we could add something like https://github.com/cgorman/tensorflow-som in the future?

lejinghu avatar Nov 12 '18 23:11 lejinghu

Hi, I still have a lot of things to add and modify that has higher priority than moving SOM to tensorflow. I don't plan it any time soon.

itdxer avatar Nov 13 '18 07:11 itdxer

Thanks. I am currently using it for data with millions of rows, but feel it's a bit slow to train. If I manage to find some ways to add GPU to Neupy implementations of SOM and GNG I will let you know.

lejinghu avatar Nov 13 '18 07:11 lejinghu

Hi, I see that you have added the ability to train on GPU: http://neupy.com/docs/algorithms/train-on-gpu.html I'm sorry, but I don't understand, is there such a possibility for GNG?

aserg24 avatar Apr 16 '19 08:04 aserg24

@aserg24 You cannot train GNG on GPU. All algorithms with fixed architecture, except RBM, cannot be trained on GPU, since they were implemented using numpy library. I'll need to update documentation in order to make it explicit

itdxer avatar Apr 16 '19 10:04 itdxer

Thank you, got it, appreciate the info.

aserg24 avatar Apr 16 '19 11:04 aserg24