첫째, 그것은 또한 CUDA 형식의 텐서의 무게를 설정해야합니다;
그리고, 상기 입력에 대응하는 크로스 엔트로피 파라미터 값의 함수로서 class_weight.
class_weight = torch.FloatTensor([0.13859937, 0.5821059, 0.63871904, 2.30220396, 7.1588294, 0]).cuda()
첫째, 그것은 또한 CUDA 형식의 텐서의 무게를 설정해야합니다;
그리고, 상기 입력에 대응하는 크로스 엔트로피 파라미터 값의 함수로서 class_weight.
class_weight = torch.FloatTensor([0.13859937, 0.5821059, 0.63871904, 2.30220396, 7.1588294, 0]).cuda()