Spaces:
Sleeping
Sleeping
Commit History
-
updated readme, ruff formatted all files
OliverPerrindf3ebbd
-
Clean up codebase and fix training bugs
OliverPerrin- Fixed total_loss tracking for validation (early stopping works now) - Lowered emotion threshold from 0.5 to 0.3 for multi-label - Optimized full.yaml config for faster training (50k samples cap) - Consolidated utils into core.py - Removed unused modules and scripts - Fixed Gutenberg download key format issue - Updated visualize_training default model name - Switch MLflow to SQLite backend1601799
-
Full training run, code cleanup, mypy/ruff fixes
OliverPerrin- Completed 3-epoch full training run (21 hours) - Emotion F1: 94.6%, Topic Accuracy: 94.2%, ROUGE-like: 0.36 - Code simplification and standardized docstrings across all modules - Fixed all mypy type errors (47 files pass) - Fixed all ruff linting errors and reformatted code - All 75 tests passing - Added GPU optimizations: TF32, Flash Attention, memory-efficient SDP - Training configs optimized for RTX 4070 12GB - tqdm progress bars for training and evaluationa47e5cf
-
updated test to mock greedy_decode correctly
OliverPerrin79c54a8
-
Refactor: Consolidate dependencies, improve testing, and add CI/CD
- Consolidated project dependencies into pyproject.toml. - Removed requirements.txt, requirements-dev.txt, and setup.py. - Removed scripts/download_data.sh. - Added comprehensive tests for src/data, src/training, and src/utils. - Fixed FutureWarning in Trainer regarding torch.amp.GradScaler. - Integrated mlflow for experiment tracking in Trainer. - Added ruff and mypy for linting and type checking. - Added .pre-commit-config.yaml for git hooks. - Added GitHub Actions CI workflow (.github/workflows/ci.yml).f6d689c