Upload 2 files

bm-general-clahe-config.yaml

@@ -12,25 +12,22 @@ dataset_json_folder: preprocessing/dataset_json_v3
 dataset_meta:
   fake:
   - create_splits: false
-    path: bitmind/ffhq-256___stable-diffusion-xl-base-1.0
-  - create_splits: false
-    path: bitmind/MS-COCO-unique___stable-diffusion-xl-base-1.0
+    name: base_transforms
+    path: bitmind/ffhq-256___stable-diffusion-xl-base-1.0_training_faces
   real:
   - create_splits: false
-    path: bitmind/ffhq-256
-  - create_splits: false
-    path: bitmind/MS-COCO-unique
+    name: base_transforms
+    path: bitmind/ffhq-256_training_faces
 ddp: false
 dry_run: false
 encoder_feat_dim: 512
-faces_only: false
 frame_num:
   test: 32
   train: 32
 lmdb: true
 lmdb_dir: ./datasets/lmdb
 local_rank: 0
-log_dir: ./logs/training/ucf_2024-09-25-04-03-26
+log_dir: ./logs/training/ucf_2024-09-25-20-29-31
 logdir: ./logs
 loss_func:
   cls_loss: cross_entropy
@@ -67,22 +64,13 @@ save_ckpt: true
 save_epoch: 1
 save_feat: true
 specific_task_number: 2
-split_transforms:
-  test:
-    name: base_transforms
-  train:
-    name: random_aug_transforms
-  validation:
-    name: base_transforms
 start_epoch: 0
 std: [0.5, 0.5, 0.5]
 test_batchSize: 32
 train_batchSize: 32
 train_dataset:
-- bitmind/ffhq-256
-- bitmind/MS-COCO-unique
-- bitmind/ffhq-256___stable-diffusion-xl-base-1.0
-- bitmind/MS-COCO-unique___stable-diffusion-xl-base-1.0
+- bitmind/ffhq-256_training_faces
+- bitmind/ffhq-256___stable-diffusion-xl-base-1.0_training_faces
 with_landmark: false
 with_mask: false
 workers: 7

bm-general-clahe.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:998045cc0cf5f2fa60b456b4eb8537e4481a3b7296c5822f2be352a89018280d
+oid sha256:afb892f2ad1a0c3417ab9a6f7a92332e819e60ef29416b5120d788a917055893
 size 188001914