Updated the config

determined-ai · Jan 29, 2024 · d38cb59 · d38cb59
1 parent 235c751
commit d38cb59
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/blog/llm-finetuning/distributed.yaml b/blog/llm-finetuning/distributed.yaml
@@ -1,13 +1,10 @@
 name: language modeling distributed
 debug: false
 environment:
-  image: determinedai/environments-dev:python-3.10-pytorch-2.0-deepspeed-0.10.0-smartsim
   environment_variables:
     - NCCL_DEBUG=INFO
-    # - HF_HOME=
 resources:
   slots_per_trial: 2
-  resource_pool: A100
 searcher:
   name: single
   max_length:
@@ -26,7 +23,7 @@ hyperparameters:
     evaluation_strategy: "steps"
     eval_steps: 1000
     logging_strategy: "steps"
-    logging_steps: 1
+    logging_steps: 100
     save_strategy: "steps"
     save_steps: 1000
     learning_rate: 1e-5