YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
HDUNet PBS
Trained by Margerie Huet Dastarac .
Training date: 06/06/2023 .
1. Task Description
Prediction of the dose distribution with proton therapy beam scanning treatment.
2. Model
2.1. Architecture
Figure 1: HDUNet PBS architecture
2.2. Input
- CT
- Target volumes contours and prescription
- Organs at risks contours
2.3. Output
- DOSE
2.4 Training details
- Number of epoch: 400
- Loss function: MSE loss
- Optimizer: AdamW
- Learning Rate: 0.0001
- Dropout: No
- Patch size in voxels: (128,128,128)
- Data augmentation used:
- RandCrop
3. Dataset
- Location: Oropharynx
- Training set size: 57
- Resolution in mm: 3x3x3
Performance
- TBD
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The model has no library tag.