#!/bin/bash
# Launch hierarchical training — Stage A (species classifier)
# Batch 1536 to utilize the 78GB free memory

cd /home/mike/Plant-Health-ID
source .venv/bin/activate
mkdir -p logs checkpoints

nohup python3 scripts/train_hierarchical.py \
	--stage species \
	--batch-size 512 \
	--no-wandb \
	>logs/species_training.log 2>&1 &

echo "Training launched (PID: $!)"
echo "Monitor with: tail -f logs/species_training.log"