Upload 4 files

face-sigmoid.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4635843775c55aab77bb1222e41fa7984016c6ebeca3c6879628c4961e34c1cf
+size 188001914

face-sigmoid.yaml

@@ -0,0 +1,76 @@
+SWA: false
+backbone_config:
+  dropout: false
+  inc: 3
+  mode: adjust_channel
+  num_classes: 2
+backbone_name: xception
+compression: c23
+cuda: true
+cudnn: true
+dataset_json_folder: preprocessing/dataset_json_v3
+dataset_meta:
+  fake:
+  - create_splits: false
+    name: base_transforms
+    path: bitmind/ffhq-256___stable-diffusion-xl-base-1.0_training_faces
+  real:
+  - create_splits: false
+    name: base_transforms
+    path: bitmind/ffhq-256_training_faces
+ddp: false
+dry_run: false
+encoder_feat_dim: 512
+frame_num:
+  test: 32
+  train: 32
+lmdb: true
+lmdb_dir: ./datasets/lmdb
+local_rank: 0
+log_dir: ./logs/training/ucf_2024-09-30-19-01-35
+logdir: ./logs
+loss_func:
+  cls_loss: cross_entropy
+  con_loss: contrastive_regularization
+  rec_loss: l1loss
+  spe_loss: cross_entropy
+losstype: null
+lr_scheduler: null
+manualSeed: 1024
+mean: [0.5, 0.5, 0.5]
+metric_scoring: auc
+mode: train
+model_name: ucf
+nEpochs: 20
+optimizer:
+  adam:
+    amsgrad: false
+    beta1: 0.9
+    beta2: 0.999
+    eps: 1.0e-08
+    lr: 0.0002
+    weight_decay: 0.0005
+  sgd:
+    lr: 0.0002
+    momentum: 0.9
+    weight_decay: 0.0005
+  type: adam
+pretrained: ../weights/xception_best.pth
+rec_iter: 100
+resolution: 256
+rgb_dir: ./datasets/rgb
+save_avg: true
+save_ckpt: true
+save_epoch: 1
+save_feat: true
+specific_task_number: 2
+start_epoch: 0
+std: [0.5, 0.5, 0.5]
+test_batchSize: 32
+train_batchSize: 32
+train_dataset:
+- bitmind/ffhq-256_training_faces
+- bitmind/ffhq-256___stable-diffusion-xl-base-1.0_training_faces
+with_landmark: false
+with_mask: false
+workers: 7

generalist-sigmoid.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6b39a7b732676043cc7b91999b3b83beb85783b6120fb838f40c7df6c0114fdb
+size 188001914

generalist-sigmoid.yaml

@@ -0,0 +1,80 @@
+SWA: false
+backbone_config:
+  dropout: false
+  inc: 3
+  mode: adjust_channel
+  num_classes: 2
+backbone_name: xception
+compression: c23
+cuda: true
+cudnn: true
+dataset_json_folder: preprocessing/dataset_json_v3
+dataset_meta:
+  fake:
+  - create_splits: false
+    path: bitmind/ffhq-256___stable-diffusion-xl-base-1.0
+  - create_splits: false
+    path: bitmind/MS-COCO-unique___stable-diffusion-xl-base-1.0
+  real:
+  - create_splits: false
+    path: bitmind/ffhq-256
+  - create_splits: false
+    path: bitmind/MS-COCO-unique
+ddp: false
+dry_run: false
+encoder_feat_dim: 512
+frame_num:
+  test: 32
+  train: 32
+lmdb: true
+lmdb_dir: ./datasets/lmdb
+local_rank: 0
+log_dir: ./logs/training/ucf_2024-09-30-01-43-34
+logdir: ./logs
+loss_func:
+  cls_loss: cross_entropy
+  con_loss: contrastive_regularization
+  rec_loss: l1loss
+  spe_loss: cross_entropy
+losstype: null
+lr_scheduler: null
+manualSeed: 1024
+mean: [0.5, 0.5, 0.5]
+metric_scoring: auc
+mode: train
+model_name: ucf
+nEpochs: 20
+optimizer:
+  adam:
+    amsgrad: false
+    beta1: 0.9
+    beta2: 0.999
+    eps: 1.0e-08
+    lr: 0.0002
+    weight_decay: 0.0005
+  sgd:
+    lr: 0.0002
+    momentum: 0.9
+    weight_decay: 0.0005
+  type: adam
+pretrained: ../weights/xception_best.pth
+rec_iter: 100
+resolution: 256
+rgb_dir: ./datasets/rgb
+save_avg: true
+save_ckpt: true
+save_epoch: 1
+save_feat: true
+specific_task_number: 2
+start_epoch: 0
+std: [0.5, 0.5, 0.5]
+test_batchSize: 32
+train_batchSize: 32
+train_dataset:
+- bitmind/ffhq-256
+- bitmind/MS-COCO-unique
+- bitmind/ffhq-256___stable-diffusion-xl-base-1.0
+- bitmind/MS-COCO-unique___stable-diffusion-xl-base-1.0
+with_landmark: false
+with_mask: false
+workers: 7