You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
423 B
21 lines
423 B
4 months ago
|
|
||
|
|
||
|
|
||
|
|
||
|
class LossManager:
|
||
|
def __init__(self):
|
||
|
pass
|
||
|
|
||
|
def compute_loss(self, outputs):
|
||
|
"""
|
||
|
计算流型损失的逻辑
|
||
|
|
||
|
:param outputs: 模型的输出
|
||
|
:return: 计算得到的流型损失值
|
||
|
"""
|
||
|
|
||
|
# 计算流型损失(这里使用均方误差作为示例)
|
||
|
manifold_loss = (outputs.abs()).mean() # 计算流型损失
|
||
|
return manifold_loss
|
||
|
|