ftnvir commited on
Commit
c54949e
·
verified ·
1 Parent(s): 3c7e685

Upload train.yaml

Browse files
Files changed (1) hide show
  1. train.yaml +62 -0
train.yaml ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: test-run
2
+ model_path: "/root/workspace/animesfw"
3
+ train_epochs: 5
4
+ batch_size: 4
5
+ resolution: 512
6
+ seed: 1138
7
+
8
+ arb:
9
+ enabled: true
10
+ debug: false
11
+ base_res: [512, 512]
12
+ max_size: [768, 512]
13
+ divisible: 64
14
+ max_ar_error: 4
15
+ min_dim: 256
16
+ dim_limit: 1024
17
+
18
+ dataset:
19
+ img_path:
20
+ - "./imagenet/imagenet_images"
21
+ egdemap_path:
22
+ - "./imagenet/imagenet_images"
23
+ captions_path:
24
+ - "./captions.json"
25
+ center_crop: true
26
+ ucg: 0.0
27
+ debug_arb: false
28
+ num_workers: 3
29
+ process_tags: false
30
+ cache_latents: false
31
+ important_tags: []
32
+ allow_duplicates: false
33
+
34
+ optimizer:
35
+ name: bitsandbytes.optim.AdamW8bit
36
+ params:
37
+ lr: 2e-4
38
+ weight_decay: 1e-2
39
+ eps: 1e-8
40
+
41
+ monitor:
42
+ wandb_id: ""
43
+ huggingface_repo: ""
44
+ huggingface_token: ""
45
+
46
+ sampling:
47
+ enabled: false
48
+ use_wandb: true
49
+ every_n_steps: -1
50
+ every_n_epochs: 1
51
+ save_dir: "samples"
52
+ seed: 1139
53
+ height: 512
54
+ width: 512
55
+ steps: 20
56
+ cfg_scale: 9
57
+ negative_prompts:
58
+ - "lowres, bad anatomy, text, error, extra digit, cropped"
59
+ - "lowres, low quality"
60
+ prompts:
61
+ - "a girl running fast on the country lane"
62
+ - "a girl in black serafuku standing in a field. Tags: solo, food, fruit, lemon, masterpiece"