pytorch tensorboard vit image-classification transformers huggingpics model-index
---
tags:
- image-classification
- pytorch
- huggingpics
metrics:
- accuracy

model-index:
- name: pneumonia-bielefeld-dl-course
  results:
  - task:
      name: Image Classification
      type: image-classification
    metrics:
      - name: Accuracy
        type: accuracy
        value: 0.8456632494926453
---

# pneumonia-bielefeld-dl-course


This registry contains the model for making pneumonia predictions and was prepared for 
Bielefeld University Deep Learning course homework.

The code used for this implementation mostly comes from here: https://github.com/nateraw/huggingpics it was a ready pipeline for model fine-tuning with huggingface and PyTorch Lightning for another dataset.
