From 812ddc87fec8bfa63e55f69534a38af18d89a0b1 Mon Sep 17 00:00:00 2001 From: Peter Soetens Date: Wed, 20 Nov 2024 12:20:32 +0100 Subject: [PATCH] dino: set num_queries to 900 to get mAP>0.5 after 12 epochs. --- notebooks/tao_launcher_starter_kit/dino/specs/evaluate.yaml | 4 ++-- notebooks/tao_launcher_starter_kit/dino/specs/export.yaml | 2 +- notebooks/tao_launcher_starter_kit/dino/specs/infer.yaml | 2 +- notebooks/tao_launcher_starter_kit/dino/specs/train.yaml | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/notebooks/tao_launcher_starter_kit/dino/specs/evaluate.yaml b/notebooks/tao_launcher_starter_kit/dino/specs/evaluate.yaml index cbf6d1c..d674668 100644 --- a/notebooks/tao_launcher_starter_kit/dino/specs/evaluate.yaml +++ b/notebooks/tao_launcher_starter_kit/dino/specs/evaluate.yaml @@ -17,7 +17,7 @@ model: num_feature_levels: 4 dec_layers: 6 enc_layers: 6 - num_queries: 300 + num_queries: 900 num_select: 100 dropout_ratio: 0.0 - dim_feedforward: 2048 \ No newline at end of file + dim_feedforward: 2048 diff --git a/notebooks/tao_launcher_starter_kit/dino/specs/export.yaml b/notebooks/tao_launcher_starter_kit/dino/specs/export.yaml index ac865b0..ce1cb30 100644 --- a/notebooks/tao_launcher_starter_kit/dino/specs/export.yaml +++ b/notebooks/tao_launcher_starter_kit/dino/specs/export.yaml @@ -12,7 +12,7 @@ model: num_feature_levels: 4 dec_layers: 6 enc_layers: 6 - num_queries: 300 + num_queries: 900 num_select: 100 dropout_ratio: 0.0 dim_feedforward: 2048 \ No newline at end of file diff --git a/notebooks/tao_launcher_starter_kit/dino/specs/infer.yaml b/notebooks/tao_launcher_starter_kit/dino/specs/infer.yaml index 6e95b79..5f90c33 100644 --- a/notebooks/tao_launcher_starter_kit/dino/specs/infer.yaml +++ b/notebooks/tao_launcher_starter_kit/dino/specs/infer.yaml @@ -21,7 +21,7 @@ model: num_feature_levels: 4 dec_layers: 6 enc_layers: 6 - num_queries: 300 + num_queries: 900 num_select: 100 dropout_ratio: 0.0 dim_feedforward: 2048 \ No newline at end of file diff --git a/notebooks/tao_launcher_starter_kit/dino/specs/train.yaml b/notebooks/tao_launcher_starter_kit/dino/specs/train.yaml index e63f8cd..39aebd7 100644 --- a/notebooks/tao_launcher_starter_kit/dino/specs/train.yaml +++ b/notebooks/tao_launcher_starter_kit/dino/specs/train.yaml @@ -27,7 +27,7 @@ model: num_feature_levels: 4 dec_layers: 6 enc_layers: 6 - num_queries: 300 + num_queries: 900 num_select: 100 dropout_ratio: 0.0 dim_feedforward: 2048