diff --git a/test_clip.sh b/test_clip.sh index 872c14d..49f0601 100644 --- a/test_clip.sh +++ b/test_clip.sh @@ -1,3 +1,5 @@ +#!/usr/bin/bash + export OMP_NUM_THREADS=2 export TORCH_DISTRIBUTED_DEBUG=INFO export TORCH_HOME="./cache/"