remove load balancing for painn tests

FAIR-Chem · Aug 6, 2024 · 62899bd · 62899bd
1 parent 22ce0c2
commit 62899bd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/core/models/test_configs/test_painn_hydra.yml b/tests/core/models/test_configs/test_painn_hydra.yml
@@ -38,7 +38,6 @@ model:
 optim:
   batch_size: 32
   eval_batch_size: 32
-  load_balancing: atoms
   eval_every: 5000
   num_workers: 2
   optimizer: AdamW