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.
20 lines
423 B
20 lines
423 B
|
|
|
|
|
|
|
|
class LossManager:
|
|
def __init__(self):
|
|
pass
|
|
|
|
def compute_loss(self, outputs):
|
|
"""
|
|
计算流型损失的逻辑
|
|
|
|
:param outputs: 模型的输出
|
|
:return: 计算得到的流型损失值
|
|
"""
|
|
|
|
# 计算流型损失（这里使用均方误差作为示例）
|
|
manifold_loss = (outputs.abs()).mean() # 计算流型损失
|
|
return manifold_loss
|
|
|
|
|