From e8b540339f3d475859118bee331d7a5ab55b9382 Mon Sep 17 00:00:00 2001
From: Xavier Bouthillier <xavier.bouthillier@umontreal.ca>
Date: Wed, 27 Nov 2024 12:55:29 -0500
Subject: [PATCH] Update standard.yaml

Update bench weights for v1.
---
 config/standard.yaml | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/config/standard.yaml b/config/standard.yaml
index f32685cbc..7deb7653a 100644
--- a/config/standard.yaml
+++ b/config/standard.yaml
@@ -31,11 +31,11 @@ convnext_large-tf32:
 
 convnext_large-tf32-fp16:
   enabled: true
-  weight: 3.0
+  weight: 1.0
 
 regnet_y_128gf:
   enabled: true
-  weight: 2.0
+  weight: 1.0
 
 bert-fp32:
   enabled: true
@@ -51,11 +51,11 @@ bert-tf32:
 
 bert-tf32-fp16:
   enabled: true
-  weight: 3.0
+  weight: 1.0
 
 t5:
   enabled: true
-  weight: 2.0
+  weight: 0.0
 
 reformer:
   enabled: true
@@ -63,11 +63,11 @@ reformer:
 
 whisper:
   enabled: true
-  weight: 1.0
+  weight: 0.0
 
 focalnet:
   enabled: true
-  weight: 2.0
+  weight: 0.0
 
 fp16:
   enabled: true
@@ -92,7 +92,7 @@ brax:
 # Diffusion
 diffusion-single:
   enabled: true
-  weight: 1.0
+  weight: 0.0
 
 diffusion-gpus:
   enabled: True
@@ -105,7 +105,7 @@ diffusion-nodes:
 # lightning
 lightning:
   enabled: true
-  weight: 1.0
+  weight: 0.0
 
 lightning-gpus:
   enabled: True
@@ -114,7 +114,7 @@ lightning-gpus:
 # dinov2
 dinov2-giant-single:
   enabled: True
-  weight: 1.0
+  weight: 0.0
 
 dinov2-giant-gpus:
   enabled: True
@@ -122,7 +122,7 @@ dinov2-giant-gpus:
 
 dinov2-giant-nodes:
   enabled: false
-  weight: 1.0
+  weight: 0.0
 
 # LLM
 llm-lora-single:
@@ -156,7 +156,7 @@ resnet152-ddp-gpus:
 # purejaxrl
 dqn:
   enabled: true
-  weight: 1.0
+  weight: 0.0
 
 ppo:
   enabled: true
@@ -164,7 +164,7 @@ ppo:
 
 cleanrljax:
   enabled: false
-  weight: 1.0
+  weight: 0.0
 
 # Geo
 dimenet:
@@ -191,7 +191,7 @@ llava-single:
 
 llava-gpus:     # This OOM
   enabled: false
-  weight: 1.0
+  weight: 0.0
 
 # rlhf
 rlhf-single:
@@ -200,7 +200,7 @@ rlhf-single:
 
 rlhf-gpus:
   enabled: true
-  weight: 1.0
+  weight: 0.0
 
 # vjepa
 vjepa-single: