Polish train_net

`train_net` accepts a device arg (that defaults to device 0) fixing #232 and upgrades V0 parameters if they're encountered (@sguada this fixes the accuracy issue you had with R-CNN). The code was slightly rinsed too.