Hugging Face's logo

Spaces:
danielzest
/
GenD-Sentinel
Configuration error

App Files Community
GenD-Sentinel
File size: 3,109 Bytes
c29babb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
from .. import config as C
from ..config import Config

experiments = {
    "example-training": [
        Config(
            backbone=C.Backbone.CLIP_L_14,
            head=C.Head.Linear,
            unfreeze_layers=["pre_layrnorm", "layer_norm1", "layer_norm2", "post_layernorm"],
            loss=C.Loss(ce_labels=1.0),
            run_dir="runs/example",
            trn_files=[
                "config/datasets/FF/test/DF.txt",
                "config/datasets/FF/test/F2F.txt",
                "config/datasets/FF/test/FS.txt",
                "config/datasets/FF/test/NT.txt",
                "config/datasets/FF/test/real.txt",
            ],
            val_files=[
                "config/datasets/FF/test/DF.txt",
                "config/datasets/FF/test/F2F.txt",
                "config/datasets/FF/test/FS.txt",
                "config/datasets/FF/test/NT.txt",
                "config/datasets/FF/test/real.txt",
            ],
            tst_files=[
                "config/datasets/CDFv2/test/Celeb-real.txt",
                "config/datasets/CDFv2/test/Celeb-synthesis.txt",
                "config/datasets/CDFv2/test/YouTube-real.txt",
            ],
            batch_size=2,
            mini_batch_size=2,
            max_epochs=1,
            wandb=False,
            devices=[0],
        )
    ],
    "example-test": [
        Config(
            run_dir="runs/test",
            tst_files=[
                "config/datasets/CDFv2/test/Celeb-real.txt",
                "config/datasets/CDFv2/test/Celeb-synthesis.txt",
                "config/datasets/CDFv2/test/YouTube-real.txt",
            ],
            batch_size=128,
            mini_batch_size=128,
            wandb=False,
            devices=[0],
        )
    ],
    "GenD_CLIP--CDFv2-example": [
        Config(
            run_dir="runs/test",
            tst_files=[
                "config/datasets/CDFv2/test/Celeb-real.txt",
                "config/datasets/CDFv2/test/Celeb-synthesis.txt",
                "config/datasets/CDFv2/test/YouTube-real.txt",
            ],
            checkpoint="yermandy/GenD_CLIP_L_14",
            max_epochs=1,
            wandb=False,
            devices=[0],
        )
    ],
    "GenD_PE--CDFv2-example": [
        Config(
            run_dir="runs/test",
            tst_files=[
                "config/datasets/CDFv2/test/Celeb-real.txt",
                "config/datasets/CDFv2/test/Celeb-synthesis.txt",
                "config/datasets/CDFv2/test/YouTube-real.txt",
            ],
            checkpoint="yermandy/GenD_PE_L",
            max_epochs=1,
            wandb=False,
            devices=[0],
        )
    ],
    "GenD_DINO--CDFv2-example": [
        Config(
            run_dir="runs/test",
            tst_files=[
                "config/datasets/CDFv2/test/Celeb-real.txt",
                "config/datasets/CDFv2/test/Celeb-synthesis.txt",
                "config/datasets/CDFv2/test/YouTube-real.txt",
            ],
            checkpoint="yermandy/GenD_DINOv3_L",
            max_epochs=1,
            wandb=False,
            devices=[0],
        )
    ],
}