EmaadKhwaja commited on
Commit
4418b4a
·
1 Parent(s): 6f8a318

file upload

Browse files
2022-04-09T07-06-25-project.yaml ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ learning_rate: 0.0003
3
+ target: dalle_main.DALLE_trainer
4
+ params:
5
+ ckpt_path: null
6
+ condition_model_path: vqgan_logs/2022-02-27T16-16-41_nucleus_vqgan/checkpoints/last.ckpt
7
+ condition_config_path: vqgan_logs/2022-02-27T16-16-41_nucleus_vqgan/configs/2022-02-27T16-16-41-project.yaml
8
+ vqgan_model_path: vqgan_logs/2022-02-27T16-17-16_threshold_vqgan/checkpoints/last.ckpt
9
+ vqgan_config_path: vqgan_logs/2022-02-27T16-17-16_threshold_vqgan/configs/2022-02-27T16-17-16-project.yaml
10
+ image_key: threshold
11
+ num_images: 2
12
+ dim: 768
13
+ num_text_tokens: 30
14
+ text_seq_len: 1000
15
+ depth: 32
16
+ heads: 16
17
+ dim_head: 64
18
+ attn_dropout: 0.1
19
+ ff_dropout: 0.1
20
+ sparse_attn: false
21
+ attn_types: full
22
+ rotary_emb: true
23
+ loss_type: logits
24
+ fixed_embedding: true
25
+ monitor: val/loss_img_epoch
26
+ text_embedding: bert
27
+ data:
28
+ target: dalle_main.CellDataModule
29
+ params:
30
+ config_file: data/train_test_split.csv
31
+ crop_size: 256
32
+ batch_size: 1
33
+ sequence_mode: embedding
34
+ vocab: bert
35
+ threshold: true
36
+ num_text_tokens: 30
37
+ text_seq_len: 1000
38
+ include_eos: true
39
+ num_workers: 16
epoch=000106.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a57231b1644c5e1cafdbe16789195d3e38455a2aa4ce46d1c0c9b6e0653f0cd
3
+ size 5037158470