switched to using (1,4,84,84) dimensional inputs, instead of minibatches. Seems it's more like what they did in deepmind