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: about 1 month 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 (2 months ago)
• Default Branch: main
• Last Pushed: 2024-12-15T10:26:57.000Z (about 2 months ago)
• Last Synced: 2024-12-15T11:26:22.366Z (about 2 months ago)
• Topics: huggingface-accelerate, python, pytorch, vision-transformer
• Language: Python
• Homepage:
• Size: 40 KB
• Stars: 0
• Watchers: 2
• 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

```