Yuechen Yang commited on
Commit
393860e
·
1 Parent(s): e1c3a18

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ There are 3 steps to tokenize the image:
25
  2. Embed each patch with a linear projection
26
  3. Each embedded patch becomes a token, and the resulting sequence of embedded patches is the sequence you pass to the model.
27
 
28
- I trained the model with 10 epochs, and I use Adam as the optimizer. The accuracy on the test set is 96%.
29
 
30
  ## Huggingface Space
31
  Huggingface space is [here](https://huggingface.co/yangy50/garbage-classification).
 
25
  2. Embed each patch with a linear projection
26
  3. Each embedded patch becomes a token, and the resulting sequence of embedded patches is the sequence you pass to the model.
27
 
28
+ I trained the model with 10 epochs, and I use Adam as the optimizer. The accuracy on the test set is 95%.
29
 
30
  ## Huggingface Space
31
  Huggingface space is [here](https://huggingface.co/yangy50/garbage-classification).