576e4

sdxs/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dca8298517571684c48f4dd35dc4892e32a6e00d5e3e191fe14698b0d9ffd812
+oid sha256:6353d1328f2909f9358a7c09e8b70bc9662d4ba22d68942c5f7c17bb6cc87f33
 size 4529095968

train.py

@@ -20,16 +20,16 @@ from torch.utils.checkpoint import checkpoint
 from diffusers.models.attention_processor import AttnProcessor2_0
 
 # --------------------------- Параметры ---------------------------
-save_path = "datasets/576p2" #"datasets/dataset384_temp" #"datasets/dataset384" #"datasets/imagenet-1kk" #"datasets/siski576" #"datasets/siski384" #"datasets/siski64" #"datasets/mnist"
-batch_size = 45 #555 #35 #7
-base_learning_rate = 2e-6 #6e-6 #2e-5 #4e-5 #3e-5 #5e-5 #8e-5
+save_path = "datasets/1152p2" #"datasets/576p2" #"datasets/dataset384_temp" #"datasets/dataset384" #"datasets/imagenet-1kk" #"datasets/siski576" #"datasets/siski384" #"datasets/siski64" #"datasets/mnist"
+batch_size = 20 #45 #555 #35 #7
+base_learning_rate = 2e-6 #8e-7 #6e-6 #2e-5 #4e-5 #3e-5 #5e-5 #8e-5
 min_learning_rate = 2.5e-5 #2e-5
-num_epochs = 8 #2 #36 #18
+num_epochs = 1 #2 #36 #18
 project = "sdxs"
-use_wandb = True
-save_model = True
+use_wandb = False
+save_model = False
 adamw8bit = True
-limit = 100000 #0
+limit = 0 #200000 #0
 checkpoints_folder = ""
 lowram = True
 use_lr_decay = False  # отключить затухание
@@ -38,7 +38,7 @@ use_lr_decay = False  # отключить затухание
 n_diffusion_steps = 40 
 samples_to_generate = 12
 guidance_scale = 5
-sample_interval_share = 10
+sample_interval_share = 20
 
 # Папки для сохранения результатов
 generated_folder = "samples"