metadata
license: apache-2.0
tags:
- pytorch
- chest-ct
- survival-analysis
- resnet
- model-2d
model-index:
- name: ResNetV2-CT
results: []
ResNetV2-CT Checkpoint
This is a PyTorch Lightning .ckpt checkpoint for a ResNetV2 model trained on chest CT images.
Usage
A quickstart script is below.
import radfusion3
dm = radfusion3.data.DataModule(config, test_split=config.test_split)
model = radfusion3.builder.build_lightning_model(config, ckpt=ckpt)
trainer.fit(model=model, datamodule=dm, ckpt_path=config.ckpt)
trainer.test(datamodule=dm, ckpt_path="best")
For detailed instructions please follow the README in Github repo.