Alan Pan

I am dealing with a multi-classification task, where mask is a two-dimensional array, and the numbers in mask represent categories, please ask me what form of mask should I pass...

Can someone share your pt file,I want to do fine tunning. My loss is too big.