kaczmarj commited on
Commit
259ba05
·
1 Parent(s): 85b8999

Upload torchscript_model.pt

Browse files

This commit adds a torchscript version of the resnet34 breast tumor classification model.

```python
import torch
import wsinfer

w = wsinfer.get_model_weights('resnet34', 'TCGA-BRCA-v1')
model = w.load_model()
c = torch.jit.script(model, example_inputs=[(torch.ones(1, 3, w.transform.resize_size, w.transform.resize_size),)])
torch.jit.save(c, root / w.url_file_name)
```

Files changed (1) hide show
  1. torchscript_model.pt +3 -0
torchscript_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2829a44b26e78ec867627fa3be7516775471577d0e80059c1af8bfc79674659a
3
+ size 85342598