Spaces:

hugging-apps
/

codecsplat

Running on Zero

App Files Files Community

codecsplat / src /loss /__init__.py

multimodalart HF Staff

CodecSplat demo

7405e85 verified 7 days ago

Raw History Blame Contribute Delete

379 Bytes

from .loss import Loss
from .loss_lpips import LossLpips, LossLpipsCfgWrapper
from .loss_mse import LossMse, LossMseCfgWrapper

LOSSES = {
    LossLpipsCfgWrapper: LossLpips,
    LossMseCfgWrapper: LossMse,
}

LossCfgWrapper = LossLpipsCfgWrapper | LossMseCfgWrapper


def get_losses(cfgs: list[LossCfgWrapper]) -> list[Loss]:
    return [LOSSES[type(cfg)](cfg) for cfg in cfgs]

	from .loss import Loss
	from .loss_lpips import LossLpips, LossLpipsCfgWrapper
	from .loss_mse import LossMse, LossMseCfgWrapper

	LOSSES = {
	LossLpipsCfgWrapper: LossLpips,
	LossMseCfgWrapper: LossMse,
	}

	LossCfgWrapper = LossLpipsCfgWrapper \| LossMseCfgWrapper


	def get_losses(cfgs: list[LossCfgWrapper]) -> list[Loss]:
	return [LOSSES[type(cfg)](cfg) for cfg in cfgs]