
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->

# swinv2-tiny-patch4-window8-256-Lego-v2-3ep

This model is a fine-tuned version of [microsoft/swinv2-tiny-patch4-window8-256](https://huggingface.co/microsoft/swinv2-tiny-patch4-window8-256) on the imagefolder dataset.
It achieves the following results on the evaluation set:
- Loss: 0.2129
- Accuracy: 0.9196

## Model description

More information needed

## Intended uses & limitations

More information needed

## Training and evaluation data

More information needed

## Training procedure

### Training hyperparameters

The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 32
- eval_batch_size: 32
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 128
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- lr_scheduler_warmup_ratio: 0.1
- num_epochs: 3

### Training results

| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
| 1.3415        | 1.0   | 49   | 0.6599          | 0.7857   |
| 0.7005        | 2.0   | 98   | 0.3036          | 0.8929   |
| 0.5615        | 3.0   | 147  | 0.2129          | 0.9196   |


### Framework versions

- Transformers 4.40.2
- Pytorch 2.2.1+cu121
- Datasets 2.19.1
- Tokenizers 0.19.1

11214 bush 3m friction with cross axle 18651 cross axle 2m with snap friction 2357 brick corner 1x2x2 3003 brick 2x2 3004 brick 1x2 3005 brick 1x1 3022 plate 2x2 3023 plate 1x2 3024 plate 1x1 3040 roof tile 1x2x45deg 3069 flat tile 1x2 32123 half bush 3673 peg 2m 3713 bush for cross axle 3794 plate 1x2 with 1 knob 6632 technic lever 3m