Skip to content

GitLab

Commit baea8882 authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 459586782
parent c0a47fd4
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/projects/detr/tasks/detection.py
View file @ baea8882
...@@ -54,6 +54,7 @@ class DectectionTask(base_task.Task): ...@@ -54,6 +54,7 @@ class DectectionTask(base_task.Task):
input_specs=input_specs, input_specs=input_specs,
backbone_config=self._task_config.model.backbone, backbone_config=self._task_config.model.backbone,
norm_activation_config=self._task_config.model.norm_activation) norm_activation_config=self._task_config.model.norm_activation)
model = detr.DETR(backbone, self._task_config.model.num_queries, model = detr.DETR(backbone, self._task_config.model.num_queries,
self._task_config.model.hidden_size, self._task_config.model.hidden_size,
self._task_config.model.num_classes, self._task_config.model.num_classes,
......
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