Closed woohee-yang closed 8 months ago
현재 multi-head 모델은 fc layer가 마스크 여부, 성별, 나이가 모두 동일하게 1792 -> 512 -> num_class 순으로 되어 있습니다. 이 부분에서 layer를 더 쌓을지, feature 개수를 다양하게 바꿀지 실험을 해봐야 할 것 같습니다.
큰 차이는 없지만 성별의 경우 클래스 개수도 다르니 나머지 둘과 파라미터 개수를 다르게 해보려고 합니다.
Background
현재 multi-head 모델은 fc layer가 마스크 여부, 성별, 나이가 모두 동일하게 1792 -> 512 -> num_class 순으로 되어 있습니다. 이 부분에서 layer를 더 쌓을지, feature 개수를 다양하게 바꿀지 실험을 해봐야 할 것 같습니다.
큰 차이는 없지만 성별의 경우 클래스 개수도 다르니 나머지 둘과 파라미터 개수를 다르게 해보려고 합니다.
Expectation