dubm commited on
Commit
f52055e
·
verified ·
1 Parent(s): 6c029dc

Upload 2 files

Browse files
Files changed (2) hide show
  1. generalist-sn34-ds.pth +3 -0
  2. 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