Skip to content

Commit

Permalink
Update bench weights for v1.
Browse files Browse the repository at this point in the history
Update bench weights for v1.
  • Loading branch information
bouthilx authored Nov 27, 2024
1 parent 63dd954 commit 1461439
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions config/standard.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -51,23 +51,23 @@ 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
weight: 1.0

whisper:
enabled: true
weight: 1.0
weight: 0.0

focalnet:
enabled: true
weight: 2.0
weight: 0.0

fp16:
enabled: true
Expand All @@ -92,7 +92,7 @@ brax:
# Diffusion
diffusion-single:
enabled: true
weight: 1.0
weight: 0.0

diffusion-gpus:
enabled: True
Expand All @@ -105,7 +105,7 @@ diffusion-nodes:
# lightning
lightning:
enabled: true
weight: 1.0
weight: 0.0

lightning-gpus:
enabled: True
Expand All @@ -114,15 +114,15 @@ lightning-gpus:
# dinov2
dinov2-giant-single:
enabled: True
weight: 1.0
weight: 0.0

dinov2-giant-gpus:
enabled: True
weight: 1.0

dinov2-giant-nodes:
enabled: false
weight: 1.0
weight: 0.0

# LLM
llm-lora-single:
Expand Down Expand Up @@ -156,15 +156,15 @@ resnet152-ddp-gpus:
# purejaxrl
dqn:
enabled: true
weight: 1.0
weight: 0.0

ppo:
enabled: true
weight: 1.0

cleanrljax:
enabled: false
weight: 1.0
weight: 0.0

# Geo
dimenet:
Expand All @@ -191,7 +191,7 @@ llava-single:

llava-gpus: # This OOM
enabled: false
weight: 1.0
weight: 0.0

# rlhf
rlhf-single:
Expand All @@ -200,7 +200,7 @@ rlhf-single:

rlhf-gpus:
enabled: true
weight: 1.0
weight: 0.0

# vjepa
vjepa-single:
Expand Down

0 comments on commit 1461439

Please sign in to comment.