Working on text2image example

Author: jloveric

config/generative_config.yaml

@@ -0,0 +1,35 @@
+files:
+  - "train-00000-of-00645-b66ac786bf6fb553.parquet"
+mlp:
+  layer_type: "polynomial"
+  n: 3
+  n_in: null
+  n_out: null
+  n_hidden: null
+  periodicity: null
+  rescale_output: False
+  scale: 2.0
+  input:
+    segments: 1
+    width: 2
+  output:
+    segments: 1
+    width: 3
+  hidden:
+    segments: 1
+    layers: 2
+    width: 10
+
+max_epochs: 50
+gpus: 1
+accelerator : cuda
+lr: 1e-3
+batch_size: 256
+
+# Are you training? Otherwise plot the result
+train: True
+checkpoint: null
+rotations: 1 # 1 is x,y, 2 is x,y, x-y, x+y, greater includes additional rotations
+
+defaults:
+  - optimizer: sparse_lion

examples/text_to_image.py

@@ -20,7 +20,7 @@
 logging.getLogger().setLevel(logging.DEBUG)
 
 
-@hydra.main(config_path="../config", config_name="images_config")
+@hydra.main(config_path="../config", config_name="generative_config")
 def run_implicit_images(cfg: DictConfig):
 
     logger.info(OmegaConf.to_yaml(cfg))
@@ -30,7 +30,7 @@ def run_implicit_images(cfg: DictConfig):
     root_dir = hydra.utils.get_original_cwd()
 
     if cfg.train is True:
-        full_path = [f"{root_dir}/{path}" for path in cfg.images]
+        full_path = [f"{root_dir}/{path}" for path in cfg.files]
         data_module = Text2ImageDataModule(
             filenames=full_path, batch_size=cfg.batch_size, rotations=cfg.rotations
         )