realfake / submit_movie.sh
devforfu
Movie stills binary classifier
c1f3687
#!/bin/bash -l
# SLURM SUBMIT SCRIPT
#SBATCH --partition=g40
#SBATCH --nodes=1
#SBATCH --gpus=8
#SBATCH --cpus-per-gpu=6
#SBATCH --job-name=realfake
#SBATCH --comment=laion
#SBATCH --signal=SIGUSR1@90
source "${HOME}/venv/bin/activate"
export NCCL_DEBUG=INFO
export PYTHONFAULTHANDLER=1
export PYTHONPATH="${HOME}/realfake"
echo "Working directory: `pwd`"
srun python3 realfake/train_cluster.py \
-jf "${HOME}/realfake/metadata/movies_plus.jsonl" \
-mn convnext_small -e=40 -fe=40 -bs=128 -wl=1 -fw=0.08 \
--acceleratorparams.devices=8 \
--acceleratorparams.strategy=ddp_find_unused_parameters_false