scripts/evaluate_hpsv2.sh
#!/bin/bash
#SBATCH --job-name=hps-eval                 # Название задачи
#SBATCH --output=hps_%j.out                 # stdout
#SBATCH --error=hps_%j.err                  # stderr
#SBATCH --time=01:00:00                     # Максимальное время выполнения
#SBATCH --nodes 1                           # Все CPU и GPU будут на одном вычислительном узле
#SBATCH --gpus 1                            # Требуемое кол-во GPU
#SBATCH --cpus-per-task 4                   # Количество выделенных ядер CPU
#SBATCH --constraint="type_e_debug"         # Предпочтительный тип узлов

module restore
source .venv/bin/activate

which python
python -V
nvidia-smi

python evaluate_hpsv2.py \
    --pretrained_model_or_path stabilityai/stable-diffusion-2-base \
    --ckpt_dir experiments/ddpo/00001-55d1-base/checkpoint-1000 \
    --out_dir experiments/ddpo/00001-55d1-base/checkpoint-1000/samples_hps \
    --batch_size 64 \
    --seed 0 \
    --skip_eval
