Update readme

Koverbay · Dec 5, 2023 · c54b4e5 · c54b4e5
1 parent 62b09f5
commit c54b4e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -139,7 +139,11 @@ do
 
   python scripts/convert_to_txt.py \
       -f "../data/two_stage_summary_text_only_t5_imgcap/${mode}_edited_csums_firststage_generated_processed.json"
+done
 
+for mode in train val test
+do
+  echo $mode
   # generate first-stage summary based on trained model
   python src/run.py \
       -model=text_only_t5 \