Skip to content

Commit

Permalink
Reduce probability of initial prompt tasks (#1764)
Browse files Browse the repository at this point in the history
The weight of initial prompt tasks for random task selection is reduced
to 0.01 which makes it very unlikely to be selected as long as other
tasks such as reply or ranking are available.

Closes: #1758
  • Loading branch information
andreaskoepf committed Feb 20, 2023
1 parent c20eb0e commit 82c9e7c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion backend/oasst_backend/tree_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,7 @@ def _random_task_selection(
task_weights[TaskType.REPLY.value] = 2

if num_missing_prompts > 0:
task_weights[TaskType.PROMPT.value] = 1
task_weights[TaskType.PROMPT.value] = 0.01

task_weights = np.array(task_weights)
weight_sum = task_weights.sum()
Expand Down

0 comments on commit 82c9e7c

Please sign in to comment.