large refactor

2022-09-25 08:31:13 +00:00
parent 84ec5a056e
commit 92891f5b26
33 changed files with 1661 additions and 1635 deletions
@@ -114,10 +114,10 @@ if __name__ == '__main__':
                        #  'uv_shaped_body': (3, 512, 256,),
                        }

-    dataset_train = factory.get_dataset(folder='/home/workspace/local-data/stylegan_new/train',
+    dataset_train = factory.get_dataset(folder='/home/workspace/local-data/train_100',
                                  batch_size=1,
                                  signature=signature_dynamic)
-    static_dataset_train = StaticDataset(folder='/home/workspace/local-data/stylegan_new/train',
+    static_dataset_train = StaticDataset(folder='/home/workspace/local-data/train_100',
                                   signature=signature_static)