Can DIGITS be used to train one network include LSTM?

NVIDIA / DIGITS

Deep Learning GPU Training System

https://developer.nvidia.com/digits

BSD 3-Clause "New" or "Revised" License

4.12k stars 1.38k forks source link

Can DIGITS be used to train one network include LSTM? #1768

Open nice-journey opened 6 years ago

nice-journey commented 6 years ago

Hi, I want to train one model which includes VGG net and LSTM module. The LSTM module is connected to VGG net to do activity recognition. Can DIGITS be used to train it? As we all known, the special point is the clip marker input. And I have no idea about the input format of clip marker to create HDF5 data format. Some open-source codes about LSTM using python layer to create input source and they are a little complicated to understand. Can you give me some advice? Thanks

lesterlo commented 6 years ago

Nvidia DIGITS support Tensorflow since version 6.0, could we use tensorflow to train LSTM in Nvidia DIGITS?