Delete generalist-sn34-ds.yaml
Browse files- generalist-sn34-ds.yaml +0 -78
generalist-sn34-ds.yaml
DELETED
@@ -1,78 +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 |
-
- path: bitmind/bm-realvisxl
|
15 |
-
- path: bitmind/bm-mobius
|
16 |
-
- path: bitmind/bm-sdxl
|
17 |
-
real:
|
18 |
-
- path: bitmind/bm-real
|
19 |
-
- path: bitmind/MS-COCO-unique
|
20 |
-
ddp: false
|
21 |
-
dry_run: false
|
22 |
-
encoder_feat_dim: 512
|
23 |
-
frame_num:
|
24 |
-
test: 32
|
25 |
-
train: 32
|
26 |
-
gpu_id: 0
|
27 |
-
lmdb: true
|
28 |
-
lmdb_dir: ./datasets/lmdb
|
29 |
-
log_dir: ./logs/training/ucf_2024-10-19-00-16-48
|
30 |
-
logdir: ./logs
|
31 |
-
loss_func:
|
32 |
-
cls_loss: cross_entropy
|
33 |
-
con_loss: contrastive_regularization
|
34 |
-
rec_loss: l1loss
|
35 |
-
spe_loss: cross_entropy
|
36 |
-
losstype: null
|
37 |
-
lr_scheduler: null
|
38 |
-
manualSeed: 1024
|
39 |
-
mean: [0.5, 0.5, 0.5]
|
40 |
-
metric_scoring: auc
|
41 |
-
mode: train
|
42 |
-
model_name: ucf
|
43 |
-
nEpochs: 5
|
44 |
-
optimizer:
|
45 |
-
adam:
|
46 |
-
amsgrad: false
|
47 |
-
beta1: 0.9
|
48 |
-
beta2: 0.999
|
49 |
-
eps: 1.0e-08
|
50 |
-
lr: 0.0002
|
51 |
-
weight_decay: 0.0005
|
52 |
-
sgd:
|
53 |
-
lr: 0.0002
|
54 |
-
momentum: 0.9
|
55 |
-
weight_decay: 0.0005
|
56 |
-
type: adam
|
57 |
-
pretrained: ../weights/xception-best.pth
|
58 |
-
rec_iter: 100
|
59 |
-
resolution: 256
|
60 |
-
rgb_dir: ./datasets/rgb
|
61 |
-
save_avg: true
|
62 |
-
save_ckpt: true
|
63 |
-
save_epoch: 1
|
64 |
-
save_feat: true
|
65 |
-
specific_task_number: 4
|
66 |
-
start_epoch: 0
|
67 |
-
std: [0.5, 0.5, 0.5]
|
68 |
-
test_batchSize: 32
|
69 |
-
train_batchSize: 32
|
70 |
-
train_dataset:
|
71 |
-
- bitmind/bm-real
|
72 |
-
- bitmind/MS-COCO-unique
|
73 |
-
- bitmind/bm-realvisxl
|
74 |
-
- bitmind/bm-mobius
|
75 |
-
- bitmind/bm-sdxl
|
76 |
-
with_landmark: false
|
77 |
-
with_mask: false
|
78 |
-
workers: 7
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|