deepsafe
/

model-code

deepfake-detection

Model card Files Files and versions

model-code / clean /video /fakestormer /scripts /vit_sbi.sh

deepsafe's picture

Add stripped inference-only model code mirror

9e14838 verified about 1 month ago

Raw History Blame Contribute Delete

299 Bytes

	#! /bin/bash

	CUDA_VISIBLE_DEVICES=0,1,2,3 python scripts/train.py --cfg configs/spatial/vit_sbi_small.yaml
	# CUDA_VISIBLE_DEVICES=0,1,2,3 python scripts/train.py --cfg configs/spatial/vit_sbi_base.yaml
	# CUDA_VISIBLE_DEVICES=0,1,2,3 python scripts/train.py --cfg configs/spatial/vit_sbi_large.yaml