GPU: AMD Radeon 8060S Graphics Memory: 133.1 GB ROCm version: 7.2.53211 ============================================================ Preparing data... ============================================================ [train] 1291202 images, 531 species [val] 234551 images, 531 species Train: 1291202 images, 2521 batches Val: 234551 images, 230 batches ============================================================ Building model... ============================================================ Model: Swin-Tiny backbone (768-dim features) Species head: 531 classes Disease heads: 521 species-specific heads ============================================================ Stage A — Species Classifier Training ============================================================ ── Sub-stage A1: Head warmup (backbone frozen) ── species_warmup: 0%| | 0/2521 [00:00