ucf_detector / config.yaml
vBogdan's picture
Upload 2 files
7909764 verified
backbone_config:
dropout: false
inc: 3
mode: adjust_channel
num_classes: 2
backbone_name: xception
compression: c23
cuda: true
cudnn: true
dataset_meta:
fake:
- path: bitmind/bm-diffusion
real:
- path: bitmind/bm-real
ddp: false
device: cuda
encoder_feat_dim: 512
frame_num:
test: 128
train: 128
gpu_id: 0
log_dir: /workspace/subnet_34/bitmind-subnet/base_miner/DFB/logs/ucf_2025-04-04-09-21-03
logdir: ./logs
loss_func:
cls_loss: cross_entropy
con_loss: contrastive_regularization
rec_loss: l1loss
spe_loss: cross_entropy
losstype: null
lr_T_max: 100
lr_eta_min: 1e-6
lr_scheduler: cosine
manualSeed: 1024
mean: [0.5, 0.5, 0.5]
metric_scoring: auc
modality: image
model_name: ucf
nEpochs: 100
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:
filename: xception-best.pth
hf_repo: bitmind/ucf
local_path: /workspace/subnet_34/bitmind-subnet/base_miner/DFB/weights/xception-best.pth
rec_iter: 100
resolution: 256
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: 128
train_batchSize: 128
train_dataset:
- bitmind/bm-real
- bitmind/bm-diffusion
with_landmark: false
with_mask: false
workers: 255