Track other files

README.md

@@ -0,0 +1,17 @@
+# SwinT-CT Checkpoint
+
+This is a PyTorch Lightning `.ckpt` checkpoint for a SwinT model trained on chest CT images.
+
+## Usage
+
+A quickstart script is below.
+
+```python
+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](https://github.com/som-shahlab/INSPECT_public/tree/main?tab=readme-ov-file#1-image-modality-experiment).

config.json

@@ -0,0 +1,6 @@
+{
+  "model_type": "model_2d",
+  "model_name": "swint_ct",
+  "pretrain_type": "supervised",
+  "num_class": 1
+}