From af99b8207000900940a59876665e8bd844cf5cb5 Mon Sep 17 00:00:00 2001
From: Elisei Rykov <e.rykov@tbank.ru>
Date: Sat, 19 Oct 2024 17:33:55 +0300
Subject: [PATCH] in progress

---
 .../train/multimodal/attention_pooling_with_n_heads.json    | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/configs/exp/train/multimodal/attention_pooling_with_n_heads.json b/configs/exp/train/multimodal/attention_pooling_with_n_heads.json
index 13c387c..8b0fcce 100644
--- a/configs/exp/train/multimodal/attention_pooling_with_n_heads.json
+++ b/configs/exp/train/multimodal/attention_pooling_with_n_heads.json
@@ -40,7 +40,7 @@
             {
                 "name": "test",
                 "records_path": "/from_s3/dataset/llava_next_data_dialogs/test_chat.jsonl",
-                "num_samples": 5000
+                "num_samples": 10000
             }
         ],
         "prompt_template": {
@@ -126,12 +126,12 @@
         "optim": "adamw_torch",
         "save_total_limit": 5,
         "load_best_model_at_end": false,
-        "deepspeed": "configs/exp/deepspeed/stage2.json",
+        "deepspeed": "/app/tutorials/deepspeed_configs/stage2.json",
         "dispatch_batches": false,
         "dataloader_num_workers": 16,
         "dataloader_pin_memory": false
     },
-    "wandb_settings": {
+    "logging_settings": {
         "project_name": "rykov",
         "run_name": "multimodal",
         "entity": "rykov"