mkshing commited on
Commit
67276b3
1 Parent(s): 0dfd194

minor fix on readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -59,7 +59,7 @@ with torch.no_grad():
59
  print("Label probs:", text_probs) # prints: [[1.0, 0.0, 0.0]]
60
  ```
61
  # Model architecture
62
- The model was trained a ViT-B/16 Transformer architecture as an image encoder and uses a 12-layer RoBERTa as a text encoder. The text encoder was trained upon the pre-trained Japanese RoBERTa model [rinna/japanese-roberta-base](https://huggingface.co/rinna/japanese-roberta-base) with the same sentencepiece tokenizer.
63
 
64
  # Training
65
  The model was trained on [CC12M](https://github.com/google-research-datasets/conceptual-12m) translated the captions to Japanese.
 
59
  print("Label probs:", text_probs) # prints: [[1.0, 0.0, 0.0]]
60
  ```
61
  # Model architecture
62
+ The model was trained a ViT-B/16 Transformer architecture as an image encoder and uses a 12-layer RoBERTa as a text encoder. The text encoder was trained upon the Japanese pre-trained RoBERTa model [rinna/japanese-roberta-base](https://huggingface.co/rinna/japanese-roberta-base) with the same sentencepiece tokenizer.
63
 
64
  # Training
65
  The model was trained on [CC12M](https://github.com/google-research-datasets/conceptual-12m) translated the captions to Japanese.