TypeError: __init__() missing 1 required positional argument: ‘coolant_config‘

···

class PhysicsLoss(tf.keras.losses.Loss):
    def __init__(self, coolant_config, reduction=tf.keras.losses.Reduction.AUTO, name="PhysicsLoss"):
        super().__init__(reduction=reduction, name=name)
        self.coolant = coolant_config
    def get_config(self):
        config = super().get_config()
        config.update({"coolant_config": self.coolant})
        return config

    @classmethod
    def from_config(cls, config):
        coolant_config = config.pop("coolant_config")
        return cls(coolant_config=coolant_config, **config)

    def call(self, y_true, y_pred):
        # 你的损失函数内容不变

在跑DNN写predictor函数的时候遇到的问题,一直有coolant_config无法识别,追溯到keras.tensorflow的loss那里,然后在自己的loss函数中声明一下就好了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值