zhangguangshanopened this issueSep 15, 2023· 1 comment zhangguangshancommentedSep 15, 2023 ms2.0镜像,进行train_dreambooth_lora脚本测试,报错:AssertionError: Only lora params 348 should be trainable. but got 628 trainable params。 SamitHuangassignedwtominSep 18, 2023...
adam = optimizers.Adam() updates = adam.get_updates(loss=K.constant(loss),constraints=[], params=self.model.trainable_weights) self.train_fn = K.function(inputs=[probs,reward], outputs=[], updates=updates) When i run the code, i get the ...