Skip to content

GitLab

Commit 8bd0c975 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Add a reference for annealing task sampler. 

PiperOrigin-RevId: 390359654
parent 0c2f6237
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
official/modeling/multitask/task_sampler.py
View file @ 8bd0c975
...@@ -78,7 +78,10 @@ class ProportionalTaskSampler(TaskSampler): ...@@ -78,7 +78,10 @@ class ProportionalTaskSampler(TaskSampler):
class AnnealingTaskSampler(TaskSampler): class AnnealingTaskSampler(TaskSampler):
"""Sample tasks according to task weights as well as training progress.""" """Sample tasks according to task weights as well as training progress.
See http://proceedings.mlr.press/v97/stickland19a/stickland19a.pdf
"""
def __init__(self, def __init__(self,
task_weights: Dict[Text, Union[float, int]], task_weights: Dict[Text, Union[float, int]],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment