Add training_args.json
Browse files- training_args.json +67 -0
training_args.json
ADDED
@@ -0,0 +1,67 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"name": "vytalbowl1",
|
3 |
+
"checkpoint": "juggernaut",
|
4 |
+
"concept_mode": "object",
|
5 |
+
"input_images": "https://res.cloudinary.com/prdg34ew78adsg/image/upload/v1718184084/user_uploads/kwwubt4yxdktdujkdgyp.jpg|https://res.cloudinary.com/prdg34ew78adsg/image/upload/v1718184090/user_uploads/saqwkbsdjnsf81xwm1sp.jpg|https://res.cloudinary.com/prdg34ew78adsg/image/upload/v1718184090/user_uploads/bd8ts7oteoafxvkzqh8k.jpg|https://res.cloudinary.com/prdg34ew78adsg/image/upload/v1718184090/user_uploads/hajts4qniguf3x1p1dzz.jpg|https://res.cloudinary.com/prdg34ew78adsg/image/upload/v1718184090/user_uploads/kugnsrxobsec3ekeehse.jpg|https://res.cloudinary.com/prdg34ew78adsg/image/upload/v1718184090/user_uploads/yaoiedslucc78rzk7rxq.jpg",
|
6 |
+
"num_training_images": 6,
|
7 |
+
"num_augmented_images": 24,
|
8 |
+
"seed": 345268398,
|
9 |
+
"resolution": 960,
|
10 |
+
"train_batch_size": 4,
|
11 |
+
"num_train_epochs": 10000,
|
12 |
+
"max_train_steps": 600,
|
13 |
+
"is_lora": true,
|
14 |
+
"prodigy_d_coef": 0.8,
|
15 |
+
"ti_lr": 0.001,
|
16 |
+
"ti_weight_decay": 0.0003,
|
17 |
+
"lora_weight_decay": 0.002,
|
18 |
+
"l1_penalty": 0.1,
|
19 |
+
"lora_param_scaler": 0.5,
|
20 |
+
"lora_rank": 12,
|
21 |
+
"snr_gamma": 5.0,
|
22 |
+
"trigger_text": "TOK",
|
23 |
+
"segmentation_prompt": "Kitchen Scene",
|
24 |
+
"crop_based_on_salience": true,
|
25 |
+
"use_face_detection_instead": false,
|
26 |
+
"clipseg_temperature": 0.6,
|
27 |
+
"left_right_flip_augmentation": true,
|
28 |
+
"augment_imgs_up_to_n": 20,
|
29 |
+
"checkpointing_steps": 10000,
|
30 |
+
"run_name": "1718184128",
|
31 |
+
"hard_pivot": false,
|
32 |
+
"off_ratio_power": 0.1,
|
33 |
+
"trainig_captions": [
|
34 |
+
"TOK is being used to prepare a meal.",
|
35 |
+
"A woman holds TOK, featuring a white bowl with a black lid.",
|
36 |
+
"TOK is clean, ready and set in the kitchen.",
|
37 |
+
"On TOK, there's an assortment of fruits and vegetables.",
|
38 |
+
"A group of pots occupy TOK.",
|
39 |
+
"A zebra logo adorns a bowl on TOK.",
|
40 |
+
"An assortment of utensils are arranged in a container on TOK.",
|
41 |
+
"Woman sitting on a couch, clutching a frisbee in her hand, is seen near TOK.",
|
42 |
+
"A view of TOK featuring a sink with a food container and water bowl.",
|
43 |
+
"Fruits and vegetables are displayed on TOK, beside a food bin.",
|
44 |
+
"TOK is home to a pan receiving its lid.",
|
45 |
+
"TOK, dimly lit, features a round bowl.",
|
46 |
+
"TOK is being used to prepare a meal.",
|
47 |
+
"A woman holds TOK, featuring a white bowl with a black lid.",
|
48 |
+
"TOK is clean, ready and set in the kitchen.",
|
49 |
+
"On TOK, there's an assortment of fruits and vegetables.",
|
50 |
+
"A group of pots occupy TOK.",
|
51 |
+
"A zebra logo adorns a bowl on TOK.",
|
52 |
+
"An assortment of utensils are arranged in a container on TOK.",
|
53 |
+
"Woman sitting on a couch, clutching a frisbee in her hand, is seen near TOK.",
|
54 |
+
"A view of TOK featuring a sink with a food container and water bowl.",
|
55 |
+
"Fruits and vegetables are displayed on TOK, beside a food bin.",
|
56 |
+
"TOK is home to a pan receiving its lid.",
|
57 |
+
"TOK, dimly lit, features a round bowl."
|
58 |
+
],
|
59 |
+
"n_training_steps": 600,
|
60 |
+
"total_n_imgs_seen": 2400,
|
61 |
+
"grid_prompts": [
|
62 |
+
"<concept>",
|
63 |
+
"<concept> as an action figure superhero, lego toy, toy story",
|
64 |
+
"a painting of <concept> by Vincent van Gogh, impressionism, oil painting, vibrant colors, texture",
|
65 |
+
"<concept> reimagined as a detailed mechanical model, complete with moving parts, metal, gears, steampunk"
|
66 |
+
]
|
67 |
+
}
|