https://github.com/touhoue/ctregression
CTRegression is a PyTorch-based project that uses Vision Transformers (ViT) and ResNet to regress the Agatston score from 3D CT scans,
https://github.com/touhoue/ctregression
huggingface-accelerate python pytorch vision-transformer
Last synced: 3 months ago
JSON representation
CTRegression is a PyTorch-based project that uses Vision Transformers (ViT) and ResNet to regress the Agatston score from 3D CT scans,
- Host: GitHub
- URL: https://github.com/touhoue/ctregression
- Owner: TouHouE
- Created: 2024-12-01T14:55:38.000Z (7 months ago)
- Default Branch: main
- Last Pushed: 2024-12-15T10:26:57.000Z (7 months ago)
- Last Synced: 2025-04-08T04:32:06.787Z (3 months ago)
- Topics: huggingface-accelerate, python, pytorch, vision-transformer
- Language: Python
- Homepage:
- Size: 40 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
## Prepare dataset & Configuration
- Modify the `config.json` under `./static`
## Training Regressor
### Multi-GPU training
```sh
export CUDA_VISIBLE_DEVICES=0,1,2...
accelerate launch main.py --gpu_ids="0,1,2..." --multi-gpu
```