File size: 1,335 Bytes
9e14838 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | class_path: src.dataset.base.ODDeepFakeDataModule
init_args:
batch_size: 20
num_workers: 4
clip_duration: 3
num_frames: 10
train_datamodules:
- class_path: src.dataset.ffpp.FFPPDataModule
init_args:
batch_size: 30
df_types: ['REAL','DF','FS','F2F','NT']
compressions: ['c23']
strategy: FORCE_PAIR
augmentations:
- NORMAL
- VIDEO
- VIDEO_RRC
- FRAME
force_random_speed: null
data_dir: 'datasets/ffpp/'
vid_ext: '.avi'
pack: false
max_clips: 3
val_datamodules:
- class_path: src.dataset.ffpp.FFPPDataModule
init_args:
df_types: ['REAL','DF','FS','F2F','NT']
compressions: ['c23']
strategy: NORMAL
augmentations:
- NONE
data_dir: 'datasets/ffpp/'
vid_ext: '.avi'
pack: false
max_clips: 1
- class_path: src.dataset.cdf.CDFDataModule
init_args:
data_dir: 'datasets/cdf/'
vid_ext: '.avi'
pack: false
max_clips: 1
- class_path: src.dataset.dfdc.DFDCDataModule
init_args:
data_dir: 'datasets/dfdc/'
vid_ext: '.avi'
pack: false
max_clips: 1
- class_path: src.dataset.fsh.FShDataModule
init_args:
compressions: ['c23']
data_dir: 'datasets/ffpp/'
vid_ext: '.avi'
pack: false
max_clips: 1 |