jungwonguk-up / KD-Font

0 stars 0 forks source link

Unet 내 num_groups parameter 통합 #33

Open gih0109 opened 11 months ago

gih0109 commented 11 months ago

개요

Unet class parameter 로 num_groups 추가 및 하위 class parameter 통합

내용

현재 num_groups parameter 는 BasicBlock class 의 parameter 에 존재하며,

ResBlock class 에만 전달되고 'TrasformerBlock` class 에는 전달되지 않는다.

이를 통합하여 Unet class 에 num_groups parameter 로 모든 group_norm 의 num_group 을 조작할 수 있도록 수정하는 것이 목표

TODO

gih0109 commented 11 months ago

수정완료 테스트