Delete config.yaml
Browse files- config.yaml +0 -88
config.yaml
DELETED
@@ -1,88 +0,0 @@
|
|
1 |
-
SWA: false
|
2 |
-
backbone_config:
|
3 |
-
dropout: false
|
4 |
-
inc: 3
|
5 |
-
mode: adjust_channel
|
6 |
-
num_classes: 2
|
7 |
-
backbone_name: xception
|
8 |
-
compression: c23
|
9 |
-
cuda: true
|
10 |
-
cudnn: true
|
11 |
-
dataset_json_folder: preprocessing/dataset_json_v3
|
12 |
-
dataset_meta:
|
13 |
-
fake:
|
14 |
-
- create_splits: false
|
15 |
-
path: bitmind/celeb-a-hq___stable-diffusion-xl-base-1.0___256_training_faces
|
16 |
-
- create_splits: false
|
17 |
-
path: bitmind/ffhq-256___stable-diffusion-xl-base-1.0_training_faces
|
18 |
-
real:
|
19 |
-
- create_splits: false
|
20 |
-
path: bitmind/celeb-a-hq_training_faces
|
21 |
-
- create_splits: false
|
22 |
-
path: bitmind/ffhq-256_training_faces
|
23 |
-
ddp: false
|
24 |
-
dry_run: false
|
25 |
-
encoder_feat_dim: 512
|
26 |
-
faces_only: true
|
27 |
-
frame_num:
|
28 |
-
test: 32
|
29 |
-
train: 32
|
30 |
-
lmdb: true
|
31 |
-
lmdb_dir: ./datasets/lmdb
|
32 |
-
local_rank: 0
|
33 |
-
log_dir: ./logs/training/ucf_2024-09-17-16-44-50
|
34 |
-
logdir: ./logs
|
35 |
-
loss_func:
|
36 |
-
cls_loss: cross_entropy
|
37 |
-
con_loss: contrastive_regularization
|
38 |
-
rec_loss: l1loss
|
39 |
-
spe_loss: cross_entropy
|
40 |
-
losstype: null
|
41 |
-
lr_scheduler: null
|
42 |
-
manualSeed: 1024
|
43 |
-
mean: [0.5, 0.5, 0.5]
|
44 |
-
metric_scoring: auc
|
45 |
-
mode: train
|
46 |
-
model_name: ucf
|
47 |
-
nEpochs: 2
|
48 |
-
optimizer:
|
49 |
-
adam:
|
50 |
-
amsgrad: false
|
51 |
-
beta1: 0.9
|
52 |
-
beta2: 0.999
|
53 |
-
eps: 1.0e-08
|
54 |
-
lr: 0.0002
|
55 |
-
weight_decay: 0.0005
|
56 |
-
sgd:
|
57 |
-
lr: 0.0002
|
58 |
-
momentum: 0.9
|
59 |
-
weight_decay: 0.0005
|
60 |
-
type: adam
|
61 |
-
pretrained: ../weights/xception_best.pth
|
62 |
-
rec_iter: 100
|
63 |
-
resolution: 256
|
64 |
-
rgb_dir: ./datasets/rgb
|
65 |
-
save_avg: true
|
66 |
-
save_ckpt: true
|
67 |
-
save_epoch: 1
|
68 |
-
save_feat: true
|
69 |
-
specific_task_number: 3
|
70 |
-
split_transforms:
|
71 |
-
test:
|
72 |
-
name: base_transforms
|
73 |
-
train:
|
74 |
-
name: random_aug_transforms
|
75 |
-
validation:
|
76 |
-
name: base_transforms
|
77 |
-
start_epoch: 0
|
78 |
-
std: [0.5, 0.5, 0.5]
|
79 |
-
test_batchSize: 32
|
80 |
-
train_batchSize: 32
|
81 |
-
train_dataset:
|
82 |
-
- bitmind/celeb-a-hq_training_faces
|
83 |
-
- bitmind/ffhq-256_training_faces
|
84 |
-
- bitmind/celeb-a-hq___stable-diffusion-xl-base-1.0___256_training_faces
|
85 |
-
- bitmind/ffhq-256___stable-diffusion-xl-base-1.0_training_faces
|
86 |
-
with_landmark: false
|
87 |
-
with_mask: false
|
88 |
-
workers: 7
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|