File size: 698 Bytes
bbb71fd 9001e40 bbb71fd e3d2f47 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
---
tags:
- pytorch_model_hub_mixin
- model_hub_mixin
datasets:
- BrachioLab/chestx
---
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
- Library: [More Information Needed]
- Docs: [More Information Needed]
## Usage
```
import torch
from exlib.datasets.chestx import ChestXDataset, ChestXPathologyModel
model = ChestXPathologyModel.from_pretrained("BrachioLab/chestx_pathols")
dataset = ChestXDataset(split="test")
dataloader = torch.utils.data.DataLoader(dataset, batch_size=4, shuffle=True)
item = next(iter(dataloader))
out = model(item["image"])
``` |