ImageNet training using torch