Downloads · 30 days
8
17% of all-time downloads
densenet_tte
StanfordShahLab/densenet_tte
densenet_tte is a machine learning model from StanfordShahLab. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
This is a PyTorch Lightning .ckpt checkpoint for a DenseNet model trained on chest CT images with TTE objective.
All-time downloads
47
Public
Repo size
45.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth45.7 MB · 100%
At a glance
- License
- apache-2.0
- Model type
- model_3d
- Access
- Public
- Created
- Apr 29, 2025
- Updated
- Apr 29, 2025
- SHA
- 4fbc58e2
Model card
From the Hugging Face model README
DenseNet Checkpoint
This is a PyTorch Lightning .ckpt checkpoint for a DenseNet model trained on chest CT images with TTE objective.
Usage
A quickstart script is below.
import torch
from src.networks import DenseNet121
model = DenseNet121(spatial_dims=3, in_channels=1, out_channels=2).to(device)
state_dict = torch.load(
loadmodel_path, map_location=f"cuda:{torch.cuda.current_device()}"
)
model.load_state_dict(state_dict)
For detailed instructions please follow the README in Github repo.
- Type
- model_3d
- License
- apache-2.0
- Created
- Apr 29, 2025
- Updated
- Apr 29, 2025