fixed train_convnet to work consistently across workstations