Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "0610b13ac4af3717f538454a9c6b1f441cb386f3"
Commit 99872555 authored by Tianqi Liu's avatar Tianqi Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 405537096
parent 491d1991
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/modeling/multitask/task_sampler.py
View file @ 99872555
...@@ -26,6 +26,10 @@ class TaskSampler(tf.Module, metaclass=abc.ABCMeta): ...@@ -26,6 +26,10 @@ class TaskSampler(tf.Module, metaclass=abc.ABCMeta):
def __init__(self, task_weights: Dict[Text, Union[float, int]]): def __init__(self, task_weights: Dict[Text, Union[float, int]]):
self._task_weights = task_weights self._task_weights = task_weights
@property
def task_weights(self):
return self._task_weights
@abc.abstractmethod @abc.abstractmethod
def task_cumulative_distribution(self, global_step: tf.Tensor) -> tf.Tensor: def task_cumulative_distribution(self, global_step: tf.Tensor) -> tf.Tensor:
"""Compute cumulative distribution to sample tasks. """Compute cumulative distribution to sample tasks.
......
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