torch.where

· Error
Pytorch를 사용하여 Siamese Network 기반 ML 모델을 구현하던 중 RuntimeError : Boolean value of Tensor with more than on values is ambiguous 에러가 발생하였다. 1. 에러 내용 Siamese Network 모델의 Train 과정 이후 Performace 측정을 위해 코드를 돌리던 중 Tensor의 값과 관련된 에러가 발생하였다. 에러가 발생한 코드는 아래와 같으며 Model의 Predict 결과를 Threshold 기준으로 하여 판단하는 부분에서 에러가 발생하였다. def evaluate_model(model, test_loader, device): model.eval() # 모델을 평가 모드로 설정 correct = 0 t..
Y30L
'torch.where' 태그의 글 목록