I saw that. The WandB report is here [0]. The losses are quite close, but in my mind we'd need to see parameter counts at least 1 and 2 params more to make conclusions (with the dataset scaling proportionally). If the training performance can be investigated, there may be some wins in this area!
2 comments
[ 2.5 ms ] story [ 18.1 ms ] thread[0] https://api.wandb.ai/links/lucidrains/lgz368mf