Implementing a deep convolutional Q-network in PyTorch