gonzalobenegas commited on
Commit
eb9c35d
·
verified ·
1 Parent(s): ccee91f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -12,6 +12,16 @@ datasets:
12
  # GPN trained on *Arabidopsis thaliana* and 7 other Brassicales
13
  See https://github.com/songlab-cal/gpn for more details.
14
 
 
 
 
 
 
 
 
 
 
 
15
  ## Some hparams:
16
  * repeat_weight: 0.1
17
  * lr: 120k at 1e-3 + 30k cosine decay
 
12
  # GPN trained on *Arabidopsis thaliana* and 7 other Brassicales
13
  See https://github.com/songlab-cal/gpn for more details.
14
 
15
+ Basic usage:
16
+ ```python
17
+ import gpn.model
18
+ from transformers import AutoModel, AutoModelForMaskedLM
19
+
20
+ model = AutoModel.from_pretrained("songlab/gpn-brassicales")
21
+ # or
22
+ model = AutoModelForMaskedLM.from_pretrained("songlab/gpn-brassicales")
23
+ ```
24
+
25
  ## Some hparams:
26
  * repeat_weight: 0.1
27
  * lr: 120k at 1e-3 + 30k cosine decay