diff --git a/config/standard.yaml b/config/standard.yaml index f32685cb..7deb7653 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: