ZepengHuo commited on
Commit
29b4be1
·
0 Parent(s):

Track ckpt with LFS

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. README.md +17 -0
  3. config.json +6 -0
  4. resnext101_ct.ckpt +3 -0
.gitattributes ADDED
@@ -0,0 +1 @@
 
 
1
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ResNeXt101-CT Checkpoint
2
+
3
+ This is a PyTorch Lightning `.ckpt` checkpoint for a ResNeXt101 model trained on chest CT images.
4
+
5
+ ## Usage
6
+
7
+ A quickstart script is below.
8
+
9
+ ```python
10
+ import radfusion3
11
+ dm = radfusion3.data.DataModule(config, test_split=config.test_split)
12
+ model = radfusion3.builder.build_lightning_model(config, ckpt=ckpt)
13
+ trainer.fit(model=model, datamodule=dm, ckpt_path=config.ckpt)
14
+ trainer.test(datamodule=dm, ckpt_path="best")
15
+ ```
16
+
17
+ 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 ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "model_2d",
3
+ "model_name": "resnext_101_ct",
4
+ "pretrain_type": "supervised",
5
+ "num_class": 1
6
+ }
resnext101_ct.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08d94075c2ed4c38b8e34e7f42ec3ddd5129f2f82ec8cf071f2b1f8200134d73
3
+ size 1042150513