Open gih0109 opened 11 months ago
Unet class parameter 로 num_groups 추가 및 하위 class parameter 통합
num_groups
현재 num_groups parameter 는 BasicBlock class 의 parameter 에 존재하며,
BasicBlock
ResBlock class 에만 전달되고 'TrasformerBlock` class 에는 전달되지 않는다.
ResBlock
이를 통합하여 Unet class 에 num_groups parameter 로 모든 group_norm 의 num_group 을 조작할 수 있도록 수정하는 것이 목표
Unet
수정완료 테스트
개요
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