EmaadKhwaja
commited on
Commit
·
4418b4a
1
Parent(s):
6f8a318
file upload
Browse files- 2022-04-09T07-06-25-project.yaml +39 -0
- epoch=000106.ckpt +3 -0
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
|