논문 구현(초분광AI) (1) 썸네일형 리스트형 [코드 분석] distillation loss 관련 내용 WHY? Distillation loss 관련 코드에서 teacher 모델과 student 모델을 통해 어떻게 loss를 계산하는지 알고 싶었습니다. HOW? - Distillation loss 는 결국 teacher모델과 student 모델 간에 코사인 유사도를 통해 loss를 계산하는 방식 같았다. (작성자 생각) WHAT? teacher 모델(z_t)은 ..?forward 코드는 다음과 같습니다.imgs는 다양한 채널 수를 가진 이미지 텐서,wave_list 는 파장 정보가 리스트 형태로 들어있습니다. def forward_zt(self, imgs, wave_list): with torch.no_grad(): if imgs.size(1)==2: .. 이전 1 다음