Upload config.yaml with huggingface_hub
Browse files- config.yaml +17 -15
config.yaml
CHANGED
|
@@ -1,31 +1,33 @@
|
|
| 1 |
-
accumulation_steps:
|
| 2 |
architecture: vit_base_patch16_224.augreg_in21k_ft_in1k
|
| 3 |
augmentations: vit_heavy
|
| 4 |
-
batch_size:
|
| 5 |
-
dataset:
|
| 6 |
epochs: 50
|
| 7 |
-
exp_name:
|
| 8 |
-
exp_path: ../runs/vit_base_patch16_224.augreg_in21k_ft_in1k-
|
| 9 |
image_size:
|
| 10 |
- 224
|
| 11 |
- 224
|
| 12 |
-
learning_rate: 0.
|
| 13 |
-
loss:
|
| 14 |
multigpu: false
|
| 15 |
number_of_classes: 2829
|
| 16 |
optimizer: SGD
|
| 17 |
random_seed: 777
|
| 18 |
root_path: ../
|
| 19 |
-
run_name: vit_base_patch16_224.augreg_in21k_ft_in1k-
|
| 20 |
scheduler: plateau
|
| 21 |
tags:
|
| 22 |
-
-
|
| 23 |
- 224x224
|
| 24 |
-
-
|
| 25 |
-
test_samples:
|
| 26 |
train: null
|
| 27 |
-
training_samples:
|
|
|
|
|
|
|
| 28 |
wandb_entity: zcu_cv
|
| 29 |
-
wandb_project:
|
| 30 |
-
wandb_run_id:
|
| 31 |
-
workers:
|
|
|
|
| 1 |
+
accumulation_steps: 4
|
| 2 |
architecture: vit_base_patch16_224.augreg_in21k_ft_in1k
|
| 3 |
augmentations: vit_heavy
|
| 4 |
+
batch_size: 128
|
| 5 |
+
dataset: FungiTastic
|
| 6 |
epochs: 50
|
| 7 |
+
exp_name: exp101
|
| 8 |
+
exp_path: ../runs/vit_base_patch16_224.augreg_in21k_ft_in1k-CrossEntropyLoss-vit_heavy/exp101
|
| 9 |
image_size:
|
| 10 |
- 224
|
| 11 |
- 224
|
| 12 |
+
learning_rate: 0.001
|
| 13 |
+
loss: CrossEntropyLoss
|
| 14 |
multigpu: false
|
| 15 |
number_of_classes: 2829
|
| 16 |
optimizer: SGD
|
| 17 |
random_seed: 777
|
| 18 |
root_path: ../
|
| 19 |
+
run_name: vit_base_patch16_224.augreg_in21k_ft_in1k-CrossEntropyLoss-vit_heavy
|
| 20 |
scheduler: plateau
|
| 21 |
tags:
|
| 22 |
+
- FungiTastic-ClosedSet
|
| 23 |
- 224x224
|
| 24 |
+
- Transformer
|
| 25 |
+
test_samples: 91832
|
| 26 |
train: null
|
| 27 |
+
training_samples: 433700
|
| 28 |
+
validation_classes: 2306
|
| 29 |
+
validation_samples: 89659
|
| 30 |
wandb_entity: zcu_cv
|
| 31 |
+
wandb_project: FungiTastic
|
| 32 |
+
wandb_run_id: npznmoww
|
| 33 |
+
workers: 12
|