#!/usr/bin/env bash
set -uo pipefail
cd /home/yurenh2/rrog; export PYTHONPATH=/home/yurenh2/rrog
echo "ESAN start gpu=${CUDA_VISIBLE_DEVICES:-?} $(date -Is)"
for s in 0 1 2 3 4; do
  echo "== esan s$s =="; python3 diag/esan_color.py --grad_mode full --seed "$s" || echo "!! esan s$s"
done
echo "doneESAN=$(date -Is)"