ok

johndpope · Oct 10, 2024 · ac4ed91 · ac4ed91
1 parent 8909c94
commit ac4ed91
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/config.yaml b/config.yaml
@@ -58,7 +58,8 @@ training:
 # Dataset parameters
 dataset:
   # celeb-hq torrent https://github.com/johndpope/MegaPortrait-hack/tree/main/junk
-  root_dir: "/media/oem/12TB/Downloads/CelebV-HQ/celebvhq/35666" # for overfitting M2Ohb0FAaJU_1.mp4 use https://github.com/johndpope/MegaPortrait-hack/tree/main/junk
+  root_dir: "./data"
+  # root_dir: "/media/oem/12TB/Downloads/CelebV-HQ/celebvhq/35666/M2Ohb0FAaJU_1" # for overfitting M2Ohb0FAaJU_1.mp4 use https://github.com/johndpope/MegaPortrait-hack/tree/main/junk
   json_file: './data/overfit.json'  # Selena Gomez
   extracted_frames: "./celebvhq/35666/images/"
   # json_file: './data/celebvhq_info.json' # 35k