ashleyradford commited on
Commit
e06c72e
1 Parent(s): e46534f

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -0
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - cats_vs_dogs
4
+ language:
5
+ - en
6
+ metrics:
7
+ - accuracy
8
+ library_name: transformers
9
+ ---
10
+
11
+ # Image Classification
12
+
13
+ Classifies cat and dog images using a subset of the cats_vs_dogs dataset.<br>
14
+ Uses PyTorch for the preprocessing, training, and inference.
15
+
16
+ ```
17
+ output_dir="cats_vs_dogs_model"
18
+ remove_unused_columns=False
19
+ evaluation_strategy="epoch"
20
+ save_strategy="epoch"
21
+ learning_rate=5e-5
22
+ per_device_train_batch_size=16
23
+ gradient_accumulation_steps=4
24
+ per_device_eval_batch_size=16
25
+ num_train_epochs=3
26
+ warmup_ratio=0.1
27
+ logging_steps=10
28
+ load_best_model_at_end=True
29
+ metric_for_best_model="accuracy"
30
+ push_to_hub=True
31
+ ```
32
+
33
+ Note: during the training, I tried adjusting some of the above hyperparameters (like making the learning rate 0.1 as we have seen in class). But then the model could only classify cats and not dogs.