|
--- |
|
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"]) |
|
``` |
