Closed jerry-ryu closed 1 year ago
random seed 고정했음에도 재현이 안되는 문제 발견
torch.use_deterministic_algorithms(True)
/torch.set_deterministic(True)
는 모델이 “deterministic” algorithms만을 사용하게 하여 재현성을 보장하는 설정이다.
EAST model은
y = F.interpolate(x[3], scale_factor=2, mode='bilinear', align_corners=True)
와 같이 비결정적으로만 실행되는 보간 or upsample layer로 구성되어 있다.
torch.use_deterministic_algorithms(True)
/torch.set_deterministic(True)
를 사용하면 upsample_bilinear2d backward cuda Runtime Error가 발생한다.
∴ EAST model은 재현적으로 학습시킬 수 없다.
다같이 데일리스크럼에서 확인했음.
Overview
Todo
재현성 보장
Additional Context