I have some number of classes problem

[LeslieLiaooo]

when i edit the n_cls=100 to 2, it gave me the error below, how could i solve it? Thanks.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [0,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [1,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [2,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [3,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [4,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [5,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [6,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [7,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [8,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [9,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [10,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [11,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [12,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [13,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [14,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [15,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [16,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [17,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [18,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [19,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [20,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [21,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [22,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [23,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [24,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [25,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [26,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [27,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [28,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [29,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [30,0,0] Assertion t >= 0 && t < n_classes failed.
C:\w\b\windows\pytorch\aten\src\ATen\native\cuda\Loss.cu:247: block: [0,0,0], thread: [31,0,0] Assertion t >= 0 && t < n_classes failed.
Traceback (most recent call last):
File "C:/Users/liaol/Desktop/ICKD/Cifar100/train_teacher.py", line 179, in
main()
File "C:/Users/liaol/Desktop/ICKD/Cifar100/train_teacher.py", line 126, in main
train_acc, train_loss = train(epoch, train_loader, model, criterion, optimizer, opt)
File "C:\Users\liaol\Desktop\ICKD\Cifar100\helper\loops.py", line 31, in train_vanilla
loss = criterion(output, target) #target
File "C:\Users\liaol\anaconda3\envs\tumor\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\liaol\anaconda3\envs\tumor\lib\site-packages\torch\nn\modules\loss.py", line 1150, in forward
return F.cross_entropy(input, target, weight=self.weight,
File "C:\Users\liaol\anaconda3\envs\tumor\lib\site-packages\torch\nn\functional.py", line 2846, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: CUDA error: device-side assert triggered