Isu dengan saiz tensor yang tidak sepadan semasa latihan dengan DeepSpeed

Saya sedang melatih model menggunakan** Pelatih Wajah** dengan penyepaduan DeepSpeed ​​dan saya menghadapi ralat yang berkaitan dengan saiz tensor yang tidak sepadan. Secara khususnya, saya mendapat ralat berikut:

Saiz tensor a (50) mesti sepadan dengan saiz tensor b (3) pada dimensi bukan tunggal2

Saya harap seseorang boleh membetulkannya dan tolong kongsikan versi anda!???

data saya:

Berikut ialah persediaan saya:

Saya telah menyemak sama ada input_id dan label mempunyai bentuk yang sama. Saya telah mengesahkan saiz kelompok dalam konfigurasi Jurulatih dan konfigurasi DeepSpeed. Saya juga telah memastikan bahawa model diletakkan dengan betul pada peranti (cuda atau cpu).

mungkin, saya tidak pasti.

Atas ialah kandungan terperinci Isu dengan saiz tensor yang tidak sepadan semasa latihan dengan DeepSpeed. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!