- Modify the
config.jsonunder./static
export CUDA_VISIBLE_DEVICES=0,1,2...
accelerate launch main.py --gpu_ids="0,1,2..." --multi-gpu-
Notifications
You must be signed in to change notification settings - Fork 0
TouHouE/CTRegression
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
CTRegression is a PyTorch-based project that uses Vision Transformers (ViT) and ResNet to regress the Agatston score from 3D CT scans,
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published