dfst / deepforge-caffe

MIT License
2 stars 3 forks source link

Remove GPU option in CaffeTrainer #134

Closed brollb closed 8 years ago

brollb commented 8 years ago

Created by: brollb

The GPU options should be removed from CaffeTrainer and should be determined on the deepforge worker (webgme executor).

This should be done by providing converting the current training prototxt to be a template that is populated by a script (bash, python or js) that checks for GPU compatibility