zxdu20's picture
Set ignore_index for CrossEntropyLoss
5c64357