Hey, are you sure that repro is exactly what you are running? When I try to run it on our master branch, I get a few runtime errors. (some are trivial, like an unqualified call to FloatTensor).
Just want to make sure we are running exactly the same code so I can look into your issue