JaeniWorld

CrossEntropyLoss를 사용 할 때 차원이 맞지 않는다고 에러가 났었다. loss_function = nn.CrossEntropyLoss() optimizer = torch.optim.Adam(model.parameters(),lr=learning_rate) loss_list = [] for epoch in range(num_epochs): for f,l in list(zip(X_train,y_train)): preds = model(f) #Fully Connected Network loss = loss_function(preds,l) 처음 코드는 이러한 형태였는데. 이 상태로 preds와 l의 size를 찍어보면 preds는 torch.Size([3]) / l은 torch.Size([])의 형..

딥러닝 모델중 CIFAR10을 사용하고 그 중 하나의 이미지 테스트 셋을 pyplot으로 시각화 할때 뜬 에러였다. 해당 코드만 돌리면 dead kernel이 뜨고 jupyter notebook이 자동으로 이를 재시작하는데 그래도 안됐다. 검색 결과 각종 라이브러리를 깔고 나서 이들끼리 충돌이 있어서 그렇다는 것을 알게 되었다. 이를 해결하기 위한 방법은 import os os.environ['KMP_DUPLICATE_LIB_OK']='True' 위의 코드를 돌린 다음 돌려야 된다. 그런 다음 pyplot이 들어있는 코드를 다시 돌리면 잘 돌아간다.