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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions