feat: Update D-FINE-N config.

Peterande · Nov 5, 2024 · a1a47c5 · a1a47c5
1 parent 063e1e6
commit a1a47c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/dfine/dfine_hgnetv2_n_coco.yml b/configs/dfine/dfine_hgnetv2_n_coco.yml
@@ -43,7 +43,7 @@ DFINETransformer:
 
   num_layers: 3
   eval_idx: -1
-  
+
   num_points: [6, 6]
 
 optimizer:
@@ -79,4 +79,4 @@ train_dataloader:
     base_size_repeat: ~
 
 val_dataloader:
-  total_batch_size: 256
+  total_batch_size: 256