Rename 2022-04-09T07-06-25-project.yaml to configg.yaml

2022-04-09T07-06-25-project.yaml → configg.yaml

@@ -2,11 +2,11 @@ model:
   learning_rate: 0.0003
   target: dalle_main.DALLE_trainer
   params:
-    ckpt_path: null
-    condition_model_path: vqgan_logs/2022-02-27T16-16-41_nucleus_vqgan/checkpoints/last.ckpt
-    condition_config_path: vqgan_logs/2022-02-27T16-16-41_nucleus_vqgan/configs/2022-02-27T16-16-41-project.yaml
-    vqgan_model_path: vqgan_logs/2022-02-27T16-17-16_threshold_vqgan/checkpoints/last.ckpt
-    vqgan_config_path: vqgan_logs/2022-02-27T16-17-16_threshold_vqgan/configs/2022-02-27T16-17-16-project.yaml
+    ckpt_path: model.ckpt
+    condition_model_path: 
+    condition_config_path: nucleus_vqgan.yaml
+    vqgan_model_path: 
+    vqgan_config_path: threshold_vqgan.yaml
     image_key: threshold
     num_images: 2
     dim: 768
@@ -22,18 +22,4 @@ model:
     rotary_emb: true
     loss_type: logits
     fixed_embedding: true
-    monitor: val/loss_img_epoch
     text_embedding: bert
-data:
-  target: dalle_main.CellDataModule
-  params:
-    config_file: data/train_test_split.csv
-    crop_size: 256
-    batch_size: 1
-    sequence_mode: embedding
-    vocab: bert
-    threshold: true
-    num_text_tokens: 30
-    text_seq_len: 1000
-    include_eos: true
-    num_workers: 16