[Bug] Reject invalid vocab-parallel CE labels - #4297

[Bug] Reject invalid vocab-parallel CE labels#4297

tianyu-l merged 1 commit into

pytorch:mainpytorch/torchtitan:mainfrom

taking-lying-flat:fix/vocab-parallel-invalid-label-gradtaking-lying-flat/torchtitan:fix/vocab-parallel-invalid-label-gradCopy head branch name to clipboard