1aurent commited on
Commit
8a54887
1 Parent(s): 8692546

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -11,9 +11,11 @@ tags:
11
  library_name: timm
12
  pipeline_tag: image-classification
13
  license: cc0-1.0
 
 
14
  ---
15
 
16
- # Model card for resnext101_32x4d.tiatoolbox-pcam
17
 
18
  A ResNeXt-D image classification model. \
19
  Trained by [Tissue Image Analytics (TIA) Centre](https://warwick.ac.uk/fac/cross_fac/tia/) on "pcam" histology patches.
@@ -48,7 +50,7 @@ img = Image.open(
48
 
49
  # load model from the hub
50
  model = timm.create_model(
51
- model_name="hf-hub:1aurent/resnext101_32x4d.tiatoolbox-pcam",
52
  pretrained=True,
53
  ).eval()
54
 
@@ -76,7 +78,7 @@ img = Image.open(
76
 
77
  # load model from the hub
78
  model = timm.create_model(
79
- model_name="hf-hub:1aurent/resnext101_32x4d.tiatoolbox-pcam",
80
  pretrained=True,
81
  num_classes=0,
82
  ).eval()
 
11
  library_name: timm
12
  pipeline_tag: image-classification
13
  license: cc0-1.0
14
+ datasets:
15
+ - 1aurent/PatchCamelyon
16
  ---
17
 
18
+ # Model card for resnext101_32x8d.tiatoolbox-pcam
19
 
20
  A ResNeXt-D image classification model. \
21
  Trained by [Tissue Image Analytics (TIA) Centre](https://warwick.ac.uk/fac/cross_fac/tia/) on "pcam" histology patches.
 
50
 
51
  # load model from the hub
52
  model = timm.create_model(
53
+ model_name="hf-hub:1aurent/resnext101_32x8d.tiatoolbox-pcam",
54
  pretrained=True,
55
  ).eval()
56
 
 
78
 
79
  # load model from the hub
80
  model = timm.create_model(
81
+ model_name="hf-hub:1aurent/resnext101_32x8d.tiatoolbox-pcam",
82
  pretrained=True,
83
  num_classes=0,
84
  ).eval()