We thus trained our GRID model with <cite>Ouchi et al. (2017)</cite> 's settings (d r = 32 and K = 8) and the best performing hyperparameters; however, we were not able to reproduce the reported gain from Grid RNN (see the row of "GRID (d r = 32, K = 8)" in Table 1 ).