label-studio-ml-backend
label-studio-ml-backend copied to clipboard
custom arguments in backend
Hello, I want to add a custom argument to the init function. But couldn't find instructions on how to pass the argument in the instantiation of the model. Here is the code which has the basic init shown and custom argument.
` class RNN_Model(LabelStudioMLBase):
def __init__(self, arg_1 = None,**kwargs):
`
Could someone point me to the instantiation of the model where I can pass custom arguments?
Hi @fsam6177 You can start your ML backend with --with keyword and params like arg_1=your_value
Hi @KonstantinKorotaev , I have a similar problem.
I added some custom parameters to the fit function, but I don't know how to pass these custom parameters in the fit function when I click on the start training button.
Any suggestions? Thank you very much!
Hi @chenshen03
You add custom param by starting your ML backend with --with keyword and params like arg_1=your_value.
If you ask about adding params from Label Studio UI - there is no such option right now, but I will create a feature request for it.