fixed the order of arguments of _update_mems() function. No impact fu…#133

victor-psiori wants to merge 1 commit intokimiyoung:masterkimiyoung/transformer-xl:masterfrom

victor-psiori:mastervictor-psiori/transformer-xl:masterCopy head branch name to clipboard