#!/usr/bin/env bash
# Single-sample overfit on UVDoc_final with OFFICIAL default hyperparameters:
#   UVDocnet, lr=2e-4, batch=8, n_epochs=10, n_epochs_decay=10,
#   alpha=beta=5, gamma=1, ep_gamma_start=10.
# Overfit branch uses deterministic crop + no aug (matches verify_ckpt_val_pipeline.py).

set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PY="${PYTHON:-/root/miniconda3/envs/o3dedit/bin/python}"
LOGDIR="${LOGDIR:-$ROOT/log_overfit_official_uvdoc}"
UV="${UV_DOC_ROOT:-$ROOT/UVDoc_final}"

exec "$PY" "$ROOT/UVDoc_official/train.py" \
  --data_to_use uvdoc \
  --data_path_UVDoc "$UV" \
  --overfit_n 1 \
  --batch_size 8 \
  --n_epochs 10 \
  --n_epochs_decay 10 \
  --lr 0.0002 \
  --alpha_w 5.0 \
  --beta_w 5.0 \
  --gamma_w 1.0 \
  --ep_gamma_start 10 \
  --num_workers "${NUM_WORKERS:-4}" \
  --device "${DEVICE:-cuda:0}" \
  --logdir "$LOGDIR"