Upload torchscript_model.pt
Browse filesThis 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)
```
- 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
|