Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
makseq committed Sep 17, 2024
1 parent 1efcf79 commit ef2c175
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ def create(cls, *args, **kwargs):
instance = super().create(*args, **kwargs)

# timeline models can be trainable and based on YOLO trained classes directly
instance.trainable = get_bool(self.control.attr, "model_trainable", "false")
instance.trainable = get_bool(instance.control.attr, "model_trainable", "false")
# if it's trainable, we need to use labels from the labeling config as is because we will train them
if instance.trainable:
instance.label_map = {label: label for label in instance.control.labels}
Expand Down

0 comments on commit ef2c175

Please sign in to comment.