add self-adaptive weighting

This commit is contained in:
giovanni
2025-08-29 19:11:48 +02:00
committed by Giovanni Canali
parent bacd7e202a
commit c42bdd575c
5 changed files with 129 additions and 0 deletions

View File

@@ -7,6 +7,7 @@ __all__ = [
"WeightingInterface",
"ScalarWeighting",
"NeuralTangentKernelWeighting",
"SelfAdaptiveWeighting",
]
from .loss_interface import LossInterface
@@ -15,3 +16,4 @@ from .lp_loss import LpLoss
from .weighting_interface import WeightingInterface
from .scalar_weighting import ScalarWeighting
from .ntk_weighting import NeuralTangentKernelWeighting
from .self_adaptive_weighting import SelfAdaptiveWeighting