테스트 코드 #Pretrained NN이 항상 1이 나오도록 학습시킨다. #Gound Truth를 1로 넣어주고 학습 시킨 후 모델 저장, 이후 test 해보면 1에 가까운 값이 나온다. #하지만 1을 넘기는 값이 나오는 문제가 발생. import torch import torch.nn as nn import torchvision.models as models from torch.autograd import Variable from torchsummary import summary from tensorboardX import SummaryWriter #cuda로 보낸다 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') print(..