如何在Keras中冻结部分权重?

时间:2019-07-24 03:36:12

标签: python tensorflow keras deep-learning

我实现了自定义正则化器my_reg。在培训期间,我希望我能拥有这样的功能:随机冻结参数的一部分,就像Dropout一样,但仅适用于此正则化器:

class my_reg(Regularizers):
    def __call__(self, loss):
        freeze_param = self.p[randomly_choose]
        freeze_param.trainable = False

如何从这些参数中随机选择并冻结它们?

0 个答案:

没有答案