StanfordShahLab/densenet_tte

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.