Upload 2 files
Browse files- bm-faces-clahe-config.yaml +13 -15
- bm-faces-clahe.pth +1 -1
bm-faces-clahe-config.yaml
CHANGED
|
@@ -12,21 +12,24 @@ dataset_json_folder: preprocessing/dataset_json_v3
|
|
| 12 |
dataset_meta:
|
| 13 |
fake:
|
| 14 |
- create_splits: false
|
| 15 |
-
path: bitmind/ffhq-256___stable-diffusion-xl-base-1.
|
|
|
|
|
|
|
| 16 |
real:
|
| 17 |
- create_splits: false
|
| 18 |
-
path: bitmind/ffhq-
|
|
|
|
|
|
|
| 19 |
ddp: false
|
| 20 |
dry_run: false
|
| 21 |
encoder_feat_dim: 512
|
| 22 |
-
faces_only: true
|
| 23 |
frame_num:
|
| 24 |
test: 32
|
| 25 |
train: 32
|
| 26 |
lmdb: true
|
| 27 |
lmdb_dir: ./datasets/lmdb
|
| 28 |
local_rank: 0
|
| 29 |
-
log_dir: ./logs/training/ucf_2024-09-
|
| 30 |
logdir: ./logs
|
| 31 |
loss_func:
|
| 32 |
cls_loss: cross_entropy
|
|
@@ -40,7 +43,7 @@ mean: [0.5, 0.5, 0.5]
|
|
| 40 |
metric_scoring: auc
|
| 41 |
mode: train
|
| 42 |
model_name: ucf
|
| 43 |
-
nEpochs:
|
| 44 |
optimizer:
|
| 45 |
adam:
|
| 46 |
amsgrad: false
|
|
@@ -63,20 +66,15 @@ save_ckpt: true
|
|
| 63 |
save_epoch: 1
|
| 64 |
save_feat: true
|
| 65 |
specific_task_number: 2
|
| 66 |
-
split_transforms:
|
| 67 |
-
test:
|
| 68 |
-
name: base_transforms
|
| 69 |
-
train:
|
| 70 |
-
name: base_transforms
|
| 71 |
-
validation:
|
| 72 |
-
name: base_transforms
|
| 73 |
start_epoch: 0
|
| 74 |
std: [0.5, 0.5, 0.5]
|
| 75 |
test_batchSize: 32
|
| 76 |
train_batchSize: 32
|
| 77 |
train_dataset:
|
| 78 |
-
- bitmind/ffhq-
|
| 79 |
-
- bitmind/
|
|
|
|
|
|
|
| 80 |
with_landmark: false
|
| 81 |
with_mask: false
|
| 82 |
-
workers:
|
|
|
|
| 12 |
dataset_meta:
|
| 13 |
fake:
|
| 14 |
- create_splits: false
|
| 15 |
+
path: bitmind/ffhq-256___stable-diffusion-xl-base-1.0
|
| 16 |
+
- create_splits: false
|
| 17 |
+
path: bitmind/MS-COCO-unique___stable-diffusion-xl-base-1.0
|
| 18 |
real:
|
| 19 |
- create_splits: false
|
| 20 |
+
path: bitmind/ffhq-256
|
| 21 |
+
- create_splits: false
|
| 22 |
+
path: bitmind/MS-COCO-unique
|
| 23 |
ddp: false
|
| 24 |
dry_run: false
|
| 25 |
encoder_feat_dim: 512
|
|
|
|
| 26 |
frame_num:
|
| 27 |
test: 32
|
| 28 |
train: 32
|
| 29 |
lmdb: true
|
| 30 |
lmdb_dir: ./datasets/lmdb
|
| 31 |
local_rank: 0
|
| 32 |
+
log_dir: ./logs/training/ucf_2024-09-25-13-30-17
|
| 33 |
logdir: ./logs
|
| 34 |
loss_func:
|
| 35 |
cls_loss: cross_entropy
|
|
|
|
| 43 |
metric_scoring: auc
|
| 44 |
mode: train
|
| 45 |
model_name: ucf
|
| 46 |
+
nEpochs: 5
|
| 47 |
optimizer:
|
| 48 |
adam:
|
| 49 |
amsgrad: false
|
|
|
|
| 66 |
save_epoch: 1
|
| 67 |
save_feat: true
|
| 68 |
specific_task_number: 2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
start_epoch: 0
|
| 70 |
std: [0.5, 0.5, 0.5]
|
| 71 |
test_batchSize: 32
|
| 72 |
train_batchSize: 32
|
| 73 |
train_dataset:
|
| 74 |
+
- bitmind/ffhq-256
|
| 75 |
+
- bitmind/MS-COCO-unique
|
| 76 |
+
- bitmind/ffhq-256___stable-diffusion-xl-base-1.0
|
| 77 |
+
- bitmind/MS-COCO-unique___stable-diffusion-xl-base-1.0
|
| 78 |
with_landmark: false
|
| 79 |
with_mask: false
|
| 80 |
+
workers: 11
|
bm-faces-clahe.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 188001914
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:922e9dbcc1dbf59ececa15e2590f367a0a958c5f0340d5dc1b71931516357f5a
|
| 3 |
size 188001914
|