--- 3. Establishing FP32 Performance Baseline ---
Evaluating:   0%|                                                                                                                                                                         | 0/5 [00:00<?, ?it/s]
Evaluating: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [01:25<00:00, 17.17s/it]
🎯 FP32 Baseline MSE Loss: 0.254275

--- 4. Wrapping model for torchao QAT ---

--- 5. Starting QAT Fine-Tuning ---
QAT Epoch 1/5:   0%|                                                                                                                                                                     | 0/44 [00:00<?, ?it/s]
QAT Epoch 1/5: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 44/44 [01:47<00:00,  2.45s/it, loss=0.319]
QAT Epoch 2/5: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 44/44 [00:01<00:00, 22.48it/s, loss=0.324]
QAT Epoch 3/5: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 44/44 [00:01<00:00, 22.91it/s, loss=0.328]
QAT Epoch 4/5: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 44/44 [00:01<00:00, 24.04it/s, loss=0.316]
QAT Epoch 5/5: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 44/44 [00:02<00:00, 21.83it/s, loss=0.325]
✅ QAT Fine-tuning complete.

--- 6. Converting, Evaluating, and Exporting torchao INT8 Model ---
Evaluating: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:00<00:00, 35.94it/s]

--- 📊 torchao QAT RESULTS ---
FP32 Baseline Loss:   0.254275
INT8 QAT Loss:        0.265800
Performance Change:   +0.011526
