Programmatically create network ?

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

Programmatically create network ? #294

Open jrabary opened 8 years ago

jrabary commented 8 years ago

Currently, I'm using DIGITS with caffe and the one of the most annoying things is the creation of the train-val.prototxt file especially if one want to design a big network. Having the ability to create the network programmaticaly would be great. An Idea is to have this program to talk with the REST API of DIGITS. Maybe this is the case with torch ?

lukeyeager commented 8 years ago

I haven't done it myself yet, but you can write a python script which creates a pycaffe network programmatically. Then you could upload that script to DIGITS in place of a "train_val" file. We'll need to do something similar to this for Torch anyway.

groar commented 8 years ago

It seems that the following haskell project does exactly that (programmatically generate prototxt as well as lua scripts) : https://github.com/ajtulloch/dnngraph

lukeyeager commented 8 years ago

Thanks for the link @groar, that's cool stuff!

/cc @jmancewicz