Add training_args.json
Browse files- training_args.json +41 -0
training_args.json
ADDED
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"name": "bb",
|
3 |
+
"mode": "concept",
|
4 |
+
"input_images": "https://eden-creations-prod.s3.us-east-1.amazonaws.com/fea8528609dd41040a55d74910b27d8c0f7f6b0d860ac6f803ba1d9321aaaf0a.jpg|https://eden-creations-prod.s3.us-east-1.amazonaws.com/a0f161929029b2fa223ff72847d1221f3cf66c4a2699c5cc031396ab469a13f0.jpg|https://eden-creations-prod.s3.us-east-1.amazonaws.com/150d577eadfa24eacc2c93e182bbfdc23adb59eff5ce496d8d0acf80d091b9ce.jpg|https://eden-creations-prod.s3.us-east-1.amazonaws.com/c0712dad4d04c7e8967ec98aaf3db266a0fbb6f828bd538788ef48c12c82ed99.jpg|https://eden-creations-prod.s3.us-east-1.amazonaws.com/3ff8f1684d425749a2eba8cdea1e827b57e9598a1b680a9e57cb816dfa627de4.jpg|https://eden-creations-prod.s3.us-east-1.amazonaws.com/dd22529ce16e804432b75adc91eb7b16e82f1196275d85d4be871f5b337df998.jpg|https://eden-creations-prod.s3.us-east-1.amazonaws.com/cfe5b944d6f7f22cb62649c3dec245bd9f85fe005f38fd4927dc633331e6279a.jpg",
|
5 |
+
"trainig_captions": [
|
6 |
+
"Illustration of a TOK sitting in front of a crowd of clowns",
|
7 |
+
"Portrait of a decorated TOK with a red rose",
|
8 |
+
"Murals of a TOK dressed in military clothing next to a sunset",
|
9 |
+
"Magician TOK with tusks and crown in front of a circus with lights",
|
10 |
+
"An old circus TOK with this is the TOK king of kingsman",
|
11 |
+
"TOKs have tusks that look like they're wearing jackets",
|
12 |
+
"Smiling TOK in a tuxedo with a circus background",
|
13 |
+
"Man holding up a phone with a TOK statue on it, in a library"
|
14 |
+
],
|
15 |
+
"num_training_images": 7,
|
16 |
+
"seed": 33157515,
|
17 |
+
"resolution": 960,
|
18 |
+
"train_batch_size": 2,
|
19 |
+
"num_train_epochs": 10000,
|
20 |
+
"max_train_steps": 800,
|
21 |
+
"is_lora": true,
|
22 |
+
"unet_learning_rate": 1e-06,
|
23 |
+
"ti_lr": 0.001,
|
24 |
+
"lora_lr": 0.0002,
|
25 |
+
"ti_weight_decay": 0.0001,
|
26 |
+
"lora_weight_decay": 0.0001,
|
27 |
+
"lora_rank": 4,
|
28 |
+
"lr_scheduler": "constant",
|
29 |
+
"lr_warmup_steps": 50,
|
30 |
+
"token_string": "TOK",
|
31 |
+
"trigger_text": "TOK",
|
32 |
+
"segmentation_prompt": "Elephant",
|
33 |
+
"crop_based_on_salience": true,
|
34 |
+
"use_face_detection_instead": false,
|
35 |
+
"clipseg_temperature": 1.0,
|
36 |
+
"left_right_flip_augmentation": true,
|
37 |
+
"checkpointing_steps": 10000,
|
38 |
+
"run_name": "1697058909",
|
39 |
+
"hard_pivot": true,
|
40 |
+
"off_ratio_power": 0.1
|
41 |
+
}
|