Setfit optimizer

someone can help with error? I want to use optimizer and scheduler but setfit does not support parameter and base_model?

Hi @Jamilah_Marram_Duart, I can see you are passing model.parameters() in the adam optimizer which is not supported. Please refer to this document to know the supported arguments for adam optimizer. And also the keras model does not have a parameters method. Thank You.

1 Like

can i use tf.keras.optimizer on setfit?

Hi @Jamilah_Marram_Duart, Yes Could you please try by

optimizer=AdamW(learning_rate=5e-5, weight_decay=0.01)

instead of using

optimizer=AdamW(model.parameters( ),learning_rate=5e-5,weight_decay=0.01)

Thank You.