I think this problem is multi classification problem. Class number is 3. This not a binary classification. why is criterion is BCEloss?