search

HumanSignal / label-studio-ml-backend

Configs and boilerplates for Label Studio's Machine Learning backend
Apache License 2.0
585 stars 261 forks source link

Button "start training" did not active fit in label-studio-ml-backend #545

Closed g811201 closed 5 months ago

g811201 commented 6 months ago

Describe the bug I use label-studio-ml backend to create a active learning loop. I would like to train model only when I click "Start Training" button. However, when I click the button, the ml-backend event in fit method get nothing rather than "START_TRAINING". I'm not sure the issue belongs to label-studio or label-studio-ml, or is there any way to achieve the goal?

To Reproduce Steps to reproduce the behavior:

  1. Go to 'Project'-->'Settings'-->'Model'-->'...'
  2. Click on 'Start Training'
  3. ML backend log in fit method get nothing

Screenshots If applicable, add screenshots to help explain your problem. image image

Environment (please complete the following information):

makseq commented 5 months ago

The problem is here: https://github.com/HumanSignal/label-studio-ml-backend/blob/master/label_studio_ml/model.py#L70

It should be START_TRAINING. Hope we will fix it soon.

https://github.com/HumanSignal/label-studio-ml-backend/pull/565

makseq commented 5 months ago

Fixed https://github.com/HumanSignal/label-studio-ml-backend/pull/565