sboshar commited on
Commit
706523a
·
verified ·
1 Parent(s): 7c2c2b7

Upload 2 files

Browse files
jax_model/hyperparams.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bucket_path": "deepchain-transformer-private",
3
+ "dataset_path": "multi-species-dataset/multi_species_final_178G.filtered.6_mers.2048_tok_len.overlap_50/1024_folds/1024_folds.p*.overlap_50.6_mers.fna",
4
+ "fixed_length": 2048,
5
+ "prepend_cls_token": true,
6
+ "num_chunks_per_worker": 128,
7
+ "batch_size": 8,
8
+ "effective_batch_size": 512,
9
+ "mixed_precision": false,
10
+ "noising_ratio": 0.15,
11
+ "masking_prob": 0.8,
12
+ "random_token_prob": 0.1,
13
+ "num_tokens_training": 300000000000,
14
+ "num_tokens_per_log": 500000000,
15
+ "num_tokens_per_validation": 5000000000,
16
+ "num_tokens_per_checkpoint": 10000000000,
17
+ "neptune_project": "Instadeep/tokenization",
18
+ "optimizer": "adamw",
19
+ "weight_decay": 0.01,
20
+ "learning_rate": 5e-05,
21
+ "schedule": true,
22
+ "warmup_steps": 16000,
23
+ "end_learning_rate": 0.0001,
24
+ "seed": 0,
25
+ "project_path": "checkpoints/",
26
+ "checkpoint_path": null
27
+ }
jax_model/pytree_ckpt.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:889877863cf47d0a2cfcf41eaafd97f91def86bfc745c41c30b3a88f68bec340
3
+ size 202906307