pytorch 예측 의 해결 여러 차례 예측 결과 불일치 문제

왜 여러 차례 예측 결과 가 일치 하지 않 습 니까?
1.매번 예측 전에 사용 하 는 지 확인

model.eval()
혹은

with torch.no_grad():
   for ...
아래 의 방법 을 추천 합 니 다.위의 방법 은 경사도 를 계산 하지만 역방향 으로 전파 되 지 않 습 니 다.아래 의 방법 은 경사도 도 계산 하지 않 고 역방향 으로 전파 되 지 않 으 며 속도 가 빠 릅 니 다.
2,모든 dropout 취소 여부 확인
3.랜 덤 피 드 설정

def setup_seed(seed):
    np.random.seed(seed)
    random.seed(seed)
    torch.manual_seed(seed) #cpu
    torch.cuda.manual_seed_all(seed)  #  gpu
    torch.backends.cudnn.deterministic = True  #cpu/gpu    
    torch.backends.cudnn.benchmark = True   #             
4.실례 화 모델 을 확보 하기 전에 istraining 을 false 로 설정 하기;이 두 줄 의 코드 순 서 는 뒤 바 꿀 수 없다.

이상 은 개인 적 인 경험 이 므 로 여러분 에 게 참고 가 되 기 를 바 랍 니 다.여러분 들 도 저 희 를 많이 응원 해 주시 기 바 랍 니 다.

좋은 웹페이지 즐겨찾기