Inheritance diagram for loss_builder.CombinedLossEvaluators:

Public Member Functions
	__init__ (self, loss_evaluators, loss_weights)

	__call__ (self, data)

Public Attributes
	loss_evaluators

	loss_weights

Detailed Description

다양한 로스의 결합을 지원하는 클래스
args:
    loss_evaluators (list): 로스 클래스 인스턴스 리스트
    loss_weights (list[float]): 각 로스별 가중치

Definition at line 12 of file loss_builder.py.

Constructor & Destructor Documentation

◆ init()

loss_builder.CombinedLossEvaluators.__init__	(	self,
		loss_evaluators,
		loss_weights )

Definition at line 19 of file loss_builder.py.

    def __init__(self, loss_evaluators, loss_weights):        
        self.loss_evaluators = loss_evaluators
        self.loss_weights = loss_weights
 

Member Function Documentation

◆ call()

loss_builder.CombinedLossEvaluators.__call__	(		self,
			data )

Definition at line 23 of file loss_builder.py.

    def __call__(self, data):        
        losses = []
        for weight, loss_evaluator in zip(self.loss_weights, self.loss_evaluators):
            losses.append(weight*loss_evaluator(data))
        
        return sum(losses)
 
 

Member Data Documentation

◆ loss_evaluators

loss_builder.CombinedLossEvaluators.loss_evaluators

Definition at line 20 of file loss_builder.py.

◆ loss_weights

loss_builder.CombinedLossEvaluators.loss_weights

Definition at line 21 of file loss_builder.py.

The documentation for this class was generated from the following file:

smrunner/losses/loss_builder.py

Public Member Functions

Public Attributes