Deepytorch Training