Bug fix. Note that convnet does not use gpu atm.