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