andrew-bitmind commited on
Commit
e0ec1c3
·
verified ·
1 Parent(s): 3f94ba0

Delete config.yaml

Browse files
Files changed (1) hide show
  1. 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