LyubomirIvanov commited on
Commit
e914ae9
·
1 Parent(s): 4f66205

Upload hyperparameters_dreamboothModel3.py

Browse files
Files changed (1) hide show
  1. hyperparameters_dreamboothModel3.py +18 -0
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