README.md

---
license: apache-2.0
tags:
- pytorch
- medical imaging
- survival analysis
- time-to-event
- resnet
- CT scans
model-index:
- name: ResNet-TTE
  results: []
---

# DenseNet Checkpoint

This is a PyTorch Lightning `.ckpt` checkpoint for a ResNet model trained on chest CT images with TTE objective.

## Usage

A quickstart script is below.

```python
import torch
from src.networks import resnet152
model = resnet152(n_input_channels=1, num_classes=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](https://github.com/som-shahlab/tte-pretraining/tree/main?tab=readme-ov-file#evaluation).