Update run_train_deepspeed.sh

mlfoundations · Oct 18, 2023 · d29c8b8 · d29c8b8
1 parent ae76178
commit d29c8b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/open_flamingo/scripts/run_train_deepspeed.sh b/open_flamingo/scripts/run_train_deepspeed.sh
@@ -35,7 +35,7 @@ srun --cpu_bind=v --accel-bind=gn bash -c 'python -u -m torch.distributed.run \
     --run_name "deepspeed" \
     --num_epochs 480 \
     --warmup_steps  0 \
-    --mmc4_textsim_threshold 0.0 \
+    --mmc4_textsim_threshold 0.24 \
     --laion_shards "/path/to/laion-samples/{000000..000001}.tar" \
     --mmc4_shards "/path/to/mmc4-samples/{000000..000001}.tar" \
     --report_to_wandb