morpheuschoo commited on
Commit
b325aa0
·
verified ·
1 Parent(s): 786f43e

Upload base_config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. base_config.yaml +36 -0
base_config.yaml ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Configuration: base_config.yaml
2
+ #
3
+ # Global training parameters. Consumed by training/train_semantic.py,
4
+ # training/train_frequency.py, and training/data/{dataset,datamodule}.py.
5
+
6
+ seed: 42
7
+ image_size: 512
8
+ batch_size: 16
9
+ num_workers: 0
10
+ lr: 0.0001
11
+
12
+ # Stream-specific settings.
13
+ semantic:
14
+ pretrained: true
15
+ freeze_backbone: true
16
+ unfreeze_last_n_blocks: 2
17
+ output_dim: 1024
18
+
19
+ frequency:
20
+ pretrained: true
21
+ freeze_backbone: false
22
+ highpass_ratio: 0.08
23
+ output_dim: 768
24
+
25
+ data:
26
+ # Path to a manifest CSV with an `image_path,label` header (0=Real, 1=AI-Generated).
27
+ # Falls back to an in-memory synthetic dataset for pipeline testing if this
28
+ # path doesn't exist yet. Produced by (see training/data/import_hf.py and
29
+ # training/data/shuffle_manifest.py):
30
+ # python -m training.data.import_hf --split train --output data/sid_subset --limit <N>
31
+ # python -m training.data.shuffle_manifest --input data/sid_subset/manifest.csv --output data/sid_subset/manifest_shuffled.csv
32
+ manifest_path: data/sid_binary_10000/manifest.csv
33
+ num_synthetic_samples: 32
34
+ val_split: 0.2
35
+
36
+ log_level: INFO