Support for multi-gpu training & evaluation

[natlouis]

Add option for DataParallel training in PyTorch. It's pretty straight forward, the only issues are when accessing the state_dict and functions belonging to model (for multi-gpu training). It becomesmodel.module.state_dict instead of model.state_dict.