Upload 2 files
Browse files- generalist-sn34-ds.pth +3 -0
- generalist-sn34-ds.yaml +78 -0
generalist-sn34-ds.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:52925d2f700151f0b213b0dc4c46670248525b7aa32d9dd61be4b94f3c9e2a69
|
3 |
+
size 188006010
|
generalist-sn34-ds.yaml
ADDED
@@ -0,0 +1,78 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|