End of training

README.md

@@ -15,28 +15,28 @@ model-index:
       value: 1.6053864712798527
       name: ForgetSet clip score of original model std (~↓)
     - type: clip
-      value: 32.02945766448975
+      value: 18.44943437576294
       name: ForgetSet clip score of learned model mean (~↑)
     - type: clip
-      value: 2.4613132609235344
+      value: 3.1584607519460643
       name: ForgetSet clip score of learned model std (~↓)
     - type: clip
-      value: 31.41413516998291
+      value: 18.025324726104735
       name: ForgetSet clip score of unlearned model mean (↓)
     - type: clip
-      value: 1.2050836697061393
+      value: 2.2593916792988926
       name: ForgetSet clip score of unlearned model std (~↓)
     - type: clip
-      value: 0.615322494506836
+      value: 0.42410964965820314
       name: ForgetSet clip score difference between learned and unlearned (↑)
     - type: clip
-      value: 2.202997475433297
+      value: 1.5386207876659266
       name: ForgetSet clip score difference between learned and unlearned std (~↓)
     - type: clip
-      value: -0.12184886932373047
+      value: 13.266961574554443
       name: ForgetSet clip score difference between original and unlearned (↑)
     - type: clip
-      value: 1.419577705662992
+      value: 2.808802361999935
       name: ForgetSet clip score difference between original and unlearned std (~↓)
     - type: clip
       value: 27.232918739318848
@@ -45,34 +45,34 @@ model-index:
       value: 3.0952564068367803
       name: RetainSet clip score of original model std (~↓)
     - type: clip
-      value: 27.819353294372558
+      value: 20.360715866088867
       name: RetainSet clip score of learned model mean (~)
     - type: clip
-      value: 2.7293971197197653
+      value: 3.1727669831548893
       name: RetainSet clip score of learned model std (~↓)
     - type: clip
-      value: 27.45349407196045
+      value: 19.10736827850342
       name: RetainSet clip score of unlearned model mean (↑)
     - type: clip
-      value: 3.7759522299495334
+      value: 3.3541435448604475
       name: RetainSet clip score of unlearned model std (~↓)
     - type: clip
-      value: 0.3658592224121094
+      value: 1.2533475875854492
       name: RetainSet clip score difference between learned and unlearned (↓)
     - type: clip
-      value: 4.175063258445924
+      value: 1.1660362978108096
       name: RetainSet clip score difference between learned and unlearned std (~↓)
     - type: clip
-      value: -0.22057533264160156
+      value: 8.125550460815429
       name: RetainSet clip score difference between original and unlearned (↓)
     - type: clip
-      value: 2.525680299775706
+      value: 3.8234689030060687
       name: RetainSet clip score difference between original and unlearned std (~↓)
     - type: runtime
-      value: 1.990412259101868
+      value: 1.938968547185262
       name: Inference latency seconds mean(↓)
     - type: runtime
-      value: 0.028140018327036147
+      value: 0.024443802025304072
       name: Inference latency seconds std(~↓)
   - task:
       type: text-to-image
@@ -82,15 +82,15 @@ model-index:
       type: forget-and-retain-together
     metrics:
     - type: runtime
-      value: 2.9832711219787598
+      value: 3.037229299545288
       name: Runtime init seconds (~↓)
     - type: runtime
-      value: 12.670677661895752
+      value: 5.76851749420166
       name: Runtime data loading seconds (~↓)
     - type: runtime
-      value: 5209.626214504242
+      value: 5229.1447513103485
       name: Runtime training seconds (↓)
     - type: runtime
-      value: 260.13896083831787
+      value: 253.97610354423523
       name: Runtime eval seconds (~↓)
 ---

checkpoint-500/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:360e5d6898c5ce628f6d2ae1f17d40ea71cce5ea4631326b1d397e564f497065
+oid sha256:488d290cf8573b311ef0458fb971487a768246468ed039bc722614e134fe3e18
 size 1722353544

checkpoint-500/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:105f1c74baae16554a5e30f9a1e577256973a7f0e193275a037c386d360a6024
+oid sha256:6f21751f52fbd797e4f4dd98e7b92943d385cde88bec982754f45bac8cb3ca86
 size 6592122

checkpoint-500/pytorch_lora_weights.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c98212d4889589baeac23cd33c293ca274f77914f73d2c9eb92751d42b62fc09
+oid sha256:01c53491f7ed574d629ab1cf4bf1390fa68c9a8b8a38e484dcae090ac0d781b1
 size 3226184

logs/text2image-fine-tune/1741021860.5716293/events.out.tfevents.1741021860.renyi.cl.itk.ppke.hu.270112.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be123ab8b0d85471838a211d51c3c115e1c76fbeebb4d13507c2d9194b3388e5
+size 2423

logs/text2image-fine-tune/1741021860.574843/hparams.yml

@@ -0,0 +1,49 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: null
+caption_column: text
+center_crop: true
+checkpointing_steps: 500
+checkpoints_total_limit: null
+dataloader_num_workers: 2
+dataset_forget_config_name: null
+dataset_forget_name: lambdalabs/naruto-blip-captions
+dataset_retain_config_name: null
+dataset_retain_name: Hamdy20002/COCO_Person
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: LeonardoBenitez/sd-lora-naruto
+hub_token: null
+image_column: image
+learning_rate: 0.0002
+local_rank: -1
+logging_dir: logs
+lr_scheduler: cosine
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 600
+mixed_precision: fp16
+noise_offset: 0
+num_train_epochs: 30
+num_validation_images: 4
+output_dir: ./assets/lora/naruto
+prediction_type: null
+pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
+push_to_hub: true
+random_flip: true
+rank: 4
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+seed: 42
+train_batch_size: 16
+use_8bit_adam: false
+validation_epochs: 3
+validation_prompt: A naruto with blue eyes.
+variant: null

logs/text2image-fine-tune/events.out.tfevents.1741021860.renyi.cl.itk.ppke.hu.270112.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2bee0073ea57f51840243930fb07afca7db39f085b3bcb9469323b96d7746791
+size 8183929

pytorch_lora_weights.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1bdd62156c1b59b37b0953be1ca089e565c983b3ac2d6b19675e7d1c750c90b0
+oid sha256:1169fa2af5f2c0fe08cd49d0e0d3e6017a4586e54d321ba0d2db574a8b0f59b5
 size 3226184