Cherry Pick from dev for imagen (#982)

firestonelib · Liujie0926 · web-flow · commit 2a41090d306f · 2022-12-01T22:05:16.000+08:00
* fix imagen training bug (#966) * fix imagen training bug * modify resolution * modify resolution Co-authored-by: Liujie0926 <44688141+Liujie0926@users.noreply.github.com> * fix imagen training bug (#966) * fix imagen training bug * modify resolution * modify resolution Co-authored-by: Liujie0926 <44688141+Liujie0926@users.noreply.github.com> * modify imagen filelist path (#981) Co-authored-by: Liujie0926 <44688141+Liujie0926@users.noreply.github.com>
diff --git a/ppfleetx/configs/multimodal/imagen/imagen_super_resolution_1024.yaml b/ppfleetx/configs/multimodal/imagen/imagen_super_resolution_1024.yaml
@@ -30,7 +30,7 @@ Data:
   Train:
     dataset:
       name: ImagenDataset
-      input_path: ./data/cc12m_base64.lst
+      input_path: ./projects/imagen/filelist/cc12m_base64.lst
       shuffle: True
       input_resolution: 1024 
       max_seq_len: 128 
diff --git a/ppfleetx/configs/multimodal/imagen/imagen_super_resolution_256.yaml b/ppfleetx/configs/multimodal/imagen/imagen_super_resolution_256.yaml
@@ -30,7 +30,7 @@ Data:
   Train:
     dataset:
       name: ImagenDataset
-      input_path: ./data/cc12m_base64.lst
+      input_path: ./projects/imagen/filelist/cc12m_base64.lst
       shuffle: True
       input_resolution: 256 
       max_seq_len: 128 
diff --git a/ppfleetx/configs/multimodal/imagen/imagen_super_resolution_512.yaml b/ppfleetx/configs/multimodal/imagen/imagen_super_resolution_512.yaml
@@ -30,7 +30,7 @@ Data:
   Train:
     dataset:
       name: ImagenDataset
-      input_path: ./data/cc12m_base64.lst
+      input_path: ./projects/imagen/filelist/cc12m_base64.lst
       shuffle: True
       input_resolution: 512 
       max_seq_len: 128 
