fn_class 질문드립니다

[elzino]

안녕하세요 Unet을 공부하는 학생입니다.

먼저 코드 공유해주셔서 너무 감사합니다! 덕분에 큰 도움이 되었습니다.

그런데 코드를 보던 중 이상한 부분이 있어 질문드립니다.

train.py에

fn_class = lambda x: 1.0 * (x > 0.5)

이 있는데, output이 logit이기 때문에

fn_class = lambda x: 1.0 * (x > 0)

으로 바뀌어야 되는것이 아닌지 질문 드립니다.

혹시 제가 잘못이해한것인지 아니면 버그인지 궁금합니다.

좋은 하루 되세요!