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.