Skip to content

train_model_multi.py : reg loss 스케일링 값 문의 #8

@KimHyeon-Ji

Description

@KimHyeon-Ji

train_model_multi.py에서 cls loss와 reg loss를 합할 때 결합 비율 외에도 reg loss에 0.00005를 곱해줘서 스케일링을 해주었는데, 해당 값도 하이퍼파라미터로 설정해주실 수 있나요?
데이터에 따라 값이 바뀌어야 할 거 같아 하이퍼파라미터로 설정되면 더 편리할 거 같습니다.
그리고 어떤 기준으로 0.00005 값을 선정했는지도 궁금합니다!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions