Replace loss.backward() in your code with accelerator.backward(loss); Gather your predictions and labels before storing them or using them for metric ...
確定! 回上一頁