Update README.md

README.md

@@ -61,6 +61,49 @@ This model was trained with the 1.4 Billion samples of the DataComp-1B dataset (
 
 **IMPORTANT NOTE:** The motivation behind dataset creation is to democratize research and experimentation around large-scale multi-modal model training and handling of uncurated, large-scale datasets crawled from publically available internet. Our recommendation is therefore to use the dataset for research purposes. Be aware that this large-scale dataset is uncurated. Keep in mind that the uncurated nature of the dataset means that collected links may lead to strongly discomforting and disturbing content for a human viewer. Therefore, please use the demo links with caution and at your own risk. It is possible to extract a “safe” subset by filtering out samples based on the safety tags (using a customized trained NSFW classifier that we built). While this strongly reduces the chance for encountering potentially harmful content when viewing, we cannot entirely exclude the possibility for harmful content being still present in safe mode, so that the warning holds also there. We think that providing the dataset openly to broad research and other interested communities will allow for transparent investigation of benefits that come along with training large-scale models as well as pitfalls and dangers that may stay unreported or unnoticed when working with closed large datasets that remain restricted to a small community. Providing our dataset openly, we however do not recommend using it for creating ready-to-go industrial products, as the basic research about general properties and safety of such large-scale models, which we would like to encourage with this release, is still in progress.
 
+## SLURM script
+
+```bash
+#!/bin/bash -x
+#SBATCH --nodes=24
+#SBATCH --gres=gpu:4
+#SBATCH --ntasks-per-node=4
+#SBATCH --cpus-per-task=12
+#SBATCH --time=24:00:00
+source /path/miniconda/bin/activate
+export CUDA_VISIBLE_DEVICES=0,1,2,3
+export MASTER_PORT=12802
+master_addr=$(scontrol show hostnames "$SLURM_JOB_NODELIST" | head -n 1)
+export MASTER_ADDR=$master_addr"i"
+echo "MASTER_ADDR="$MASTER_ADDR
+srun --cpu-bind=v --cpus-per-task=12 python -u -m training.main --aug-cfg scale='(0.4, 1.0)' color_jitter='(0.32, 0.32, 0.32, 0.08)' color_jitter_prob=0.8 gray_scale_prob=0.2 use_timm=True \
+--save-frequency 1 \
+--zeroshot-frequency 1 \
+--dataset-type webdataset \
+--train-data '/path/to/data' \
+--report-to tensorboard \
+--train-num-samples 1398270000 \
+--warmup 2000 \
+--batch-size 896 \
+--epochs 24 \
+--workers 8 \
+--model ViT-B-32-256 \
+--logs logs \
+--seed 0 \
+--ddp-static-graph \
+--local-loss \
+--gather-with-grad \
+--lr 0.001 \
+--log-every-n-steps 20 \
+--save-most-recent \
+--resume latest \
+--grad-checkpointing \
+--precision amp_bfloat16 \
+--beta1 0.9 \
+--beta2 0.95 \
+--wd 0.2
+```
+
 # Evaluation
 
 Evaluation done on 38 datasets, using [LAION CLIP Benchmark](https://github.com/LAION-AI/CLIP_benchmark).