Fixed indices device bug for W6 models (train_aux.py)#1858

Open

stemann wants to merge 1 commit intoWongKinYiu:mainfrom IHPSystems:feature/fix_loss_device_bug_for_train_aux

+2-2

Commits on Sep 19, 2023

Fixed indices device bug for W6 models (train_aux.py)
WongKinYiu
committed