loss_builder Namespace Reference

Classes
class	CombinedLossEvaluators

Functions
	build_loss (cfg)

Function Documentation

build_loss()

loss_builder.build_loss

(

cfg

)

모델의 학습에 사용할 로스 계산기를 빌드하는 기능
args:
    cfg (str or Config): 학습 파라미터가 정의된 모델의 config 파일 경로 또는 mmengine.config.Config.fromfile()로 생성한 객체

Definition at line 31 of file loss_builder.py.

def build_loss(cfg):
    """
    모델의 학습에 사용할 로스 계산기를 빌드하는 기능
    args:
        cfg (str or Config): 학습 파라미터가 정의된 모델의 config 파일 경로 또는 mmengine.config.Config.fromfile()로 생성한 객체
    """
 
    #config 로드
    if isinstance(cfg, str):
        cfg = Config.fromfile(cfg_path)
 
    loss_evaluators = []
    loss_weights = []
 
    losses = cfg.loss #로스 파라미터
    loss_types = losses.keys() #로스 클래스 목록
 
    for type in loss_types:
 
        #예외처리
        assert type in __loss_builder__, \
            f'not found loss type : {type}'
 
        loss_cfg = losses.get(type) #losses[type]
        loss_weights.append(loss_cfg.weight) #로스에 대한 가중치
        loss_cfg.pop('weight')
        loss_evaluators.append(__loss_builder__[type](**loss_cfg)) #로스 클래스 빌드
        
    #로스의 결합 클래스 빌드
    return CombinedLossEvaluators(loss_evaluators, loss_weights)