Alston Lo

I also had the same question. Based on the discussion from lucidrains/x-transformers#40, it seems that doing a partial rotation can slightly improve performance. I think it could help with clearing...

Only semi-related to the current issue, but rerunning the same script with the nightly build (as of now) raises an error. This is due to the local `tpu` variable in...

Thanks! @awaelchli I have installed lightning directly from #18085 and commented out the suggested lines. The training script runs but no WandB run is ever created and nothing is logged...

While trying to find a solution for this issue, I think I may have stumbled upon another potential bug (which I suspect may be causing this issue, but I am...