Commit ·
e914ae9
1
Parent(s): 4f66205
Upload hyperparameters_dreamboothModel3.py
Browse files
hyperparameters_dreamboothModel3.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
instance_prompt="a sks dog"
|
| 2 |
+
class_prompt="a dog"
|
| 3 |
+
num_inference_steps=50
|
| 4 |
+
guidance_scale=7.5
|
| 5 |
+
train_batch_size=1
|
| 6 |
+
learning_rate="2e-6"
|
| 7 |
+
lr_scheduler="linear"
|
| 8 |
+
model_id_finetuned = "/content/model"
|
| 9 |
+
model_id_pretrained = "stabilityai/stable-diffusion-2-base"
|
| 10 |
+
num_class_images=200
|
| 11 |
+
train_steps=600
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
sample_output_dir="/content/samples"
|
| 15 |
+
model_output_dir="/content/model"
|
| 16 |
+
class_data_dir="/content/resized_class"
|
| 17 |
+
instance_data_dir="/content/resized_instance"
|
| 18 |
+
image_size=512
|