This commit is contained in:
Craig
2025-04-12 10:28:50 +01:00
parent ae79a555d3
commit c3096f0664
3 changed files with 53 additions and 5 deletions

View File

@@ -16,10 +16,7 @@ This list outlines the steps required to complete the Torchvision Finetuning pro
- [x] Install pre-commit hooks
- [x] Verify PyTorch GPU integration (`scripts/check_gpu.py`)
- [x] Create data download script (`scripts/download_data.sh`)
- [ ] Implement PennFudanDataset class (`utils/data_utils.py`)
- [ ] Implement model finetuning logic (`models/detection.py`)
- [ ] Implement training script (`train.py`)
- [ ] Implement evaluation script (`test.py`)
- [x] Implement configuration system (`configs/base_config.py`, `configs/pennfudan_maskrcnn_config.py`)
## Phase 2: Data Handling & Model
@@ -125,4 +122,4 @@ This list outlines the steps required to complete the Torchvision Finetuning pro
- [ ] Dependencies list
- [ ] (Optional) Results section
- [ ] Perform final code quality checks (`ruff format .`, `ruff check . --fix`).
- [ ] Ensure all pre-commit hooks pass.
- [ ] Ensure all pre-commit hooks pass.