Internal change

PiperOrigin-RevId: 405537096

Internal change
PiperOrigin-RevId: 405537096
99872555 · Tianqi Liu · A. Unique TensorFlower · 491d1991 · 99872555
Commit 99872555 authored Oct 25, 2021 by Tianqi Liu Committed by A. Unique TensorFlower Oct 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

official/modeling/multitask/task_sampler.py official/modeling/multitask/task_sampler.py +4 -0

No files found.
--- a/official/modeling/multitask/task_sampler.py
+++ b/official/modeling/multitask/task_sampler.py
@@ -26,6 +26,10 @@ class TaskSampler(tf.Module, metaclass=abc.ABCMeta):
  def __init__(self, task_weights: Dict[Text, Union[float, int]]):
    self._task_weights = task_weights

+  @property
+  def task_weights(self):
+    return self._task_weights
+
  @abc.abstractmethod
  def task_cumulative_distribution(self, global_step: tf.Tensor) -> tf.Tensor:
    """Compute cumulative distribution to sample tasks.