File size: 829 Bytes
42183da 3aeea03 42183da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Script for fine-tuning our FSFM-3C model for evaluation of cross-dataset deepfakes detection:
```
CUDA_VISIBLE_DEVICES=0,1 OMP_NUM_THREADS=1 python -m torch.distributed.launch --node_rank=0 --nproc_per_node=2 main_finetune_DfD.py \
--accum_iter 1 \
--apply_simple_augment \
--batch_size 32 \
--nb_classes 2 \
--model vit_base_patch16 \
--epochs 10 \
--blr 2.5e-4 \
--layer_decay 0.65 \
--weight_decay 0.05 \
--drop_path 0.1 \
--reprob 0.25 \
--mixup 0.8 \
--cutmix 1.0 \
--dist_eval \
--finetune ../../pretrain/checkpoint/pretrained_models/VF2_ViT-B/checkpoint-400.pth \
--finetune_data_path ../../../datasets/finetune_datasets/deepfakes_detection/FaceForensics/32_frames/DS_FF++_all_cls/c23 \
--output_dir ./checkpoint/finetuned_models/FF++_c23_32frames
``` |