metadata

license: apache-2.0
tags:
  - pytorch
  - resnext
  - chest-ct
  - survival-analysis
  - time-to-event
  - model-3d
model-index:
  - name: resnext
    results: []

ResNeXt101-CT Checkpoint

This is a PyTorch Lightning .ckpt checkpoint for a ResNeXt101 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.