File size: 331 Bytes
1dc2504
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
#!/usr/bin/env bash
set -euo pipefail

export WANDB_PROJECT=enhanced-deepfake-detector
export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"

python3 -m src.train.train --config configs/train/aat_pgd.yaml
python3 -m src.eval.evaluate --checkpoint outputs/best.pt --config configs/train/aat_pgd.yaml
python3 -m src.eval.ablation