hushon / noisy-label

12 stars 0 forks source link

NoisyCIFAR 데이터셋 randomness 문제 #10

Closed hushon closed 1 year ago

hushon commented 1 year ago

NoisyCIFAR 생성 시마다 noisy label 이 달라지도록 구현되어 있어서 모델 학습시 training label 과 벤치마크시 training label 이 불일치하기 때문에 필터링 성능이 잘못 측정되고 있었음.

hushon commented 1 year ago

현재 torch.distributions.Categorical() 는 random number generator 를 컨트롤하지 못하도록 구현되어 있음. 구현을 바꿔서 샘플링시 torch.Generator() 를 입력받을 수 있도록 변경.