A Professional PyTorch Training Loop — Training the Network — Atlas