CNN LSTM machine learning model on GPU?

TMVA
1

Hello,

I have found this talk (slide 14) from 5 years ago and I am particularly willing to use CNN-LSTM machine learning model on GPU. It seems this wasn’t implemented yet at that time. Were there some recent development about LSTM model for TMVA ?

Screenshot 2023-02-24 at 03.56.44
Screenshot 2023-02-24 at 03.56.441176×970 27.8 KB

Thanks !

2

That’s a question for @moneta I suppose

3

If needed I can surely develop some code if this is not too difficult task and if documentation is available

I have also found that it is possible to import model developed with python ml packages (Keras,etc…). How CPU/GPU featured are interfaced ?
Is it restricted to PyRoot or can this be used also with C++ root ? (I am only using C++)

4

As I said, @moneta will most probably guide you

5

Hi,

The CNN and recurrent including LSTM and GRU are all supported on GPU using the native Deep Learning package. You can see example on how to use them in the tutorial
https://root.cern.ch/doc/master/TMVA__CNN__Classification_8C.html

and

https://root.cern.ch/doc/master/TMVA__RNN__Classification_8C.html

The GPU is automatically enabled if TMVA has been built with GPU support (-Dtmva-gpu=On -Dcuda=On).
The tutorials show also how you can train the model using Keras or PyTorch using the TMVA interface.

Best,

Lorenzo