Model: LeonardoBenitez/demo-vision-unlearning-munba
Author: LeonardoBenitez
Tags: tensorboard, model-index, region:us
Card metadata: hyperparameters: lora_r: 4
lora_alpha: 4
is_lora_negated: yes
seed: 42
model_name_or_path: CompVis/stable-diffusion-v1-4
dataset_forget_name: assets/datasets/imagenette_splits/n03028079/train_forget
dataset_retain_name: assets/datasets/imagenette_splits/n03028079/train_retain
image_column: image
caption_column: text
validation_prompt: An image of a church
num_validation_images: 1
validation_epochs: 1
resolution: 512
center_crop: no
random_flip: yes
dataloader_num_workers: 2
per_device_train_batch_size: 4
gradient_accumulation_steps: 1
num_train_epochs: 2
learning_rate: 0.0002
lr_scheduler_type: constant
should_log: yes
local_rank: -1
device: cuda
n_gpu: 1
output_dir: assets/models/munba_church_002
hub_model_id: LeonardoBenitez/demo-vision-unlearning-munba
logging_dir: logs
logging_steps: 20
save_strategy: epoch
save_total_limit: 2
gradient_checkpointing: no
enable_xformers_memory_efficient_attention: no
mixed_precision: no
allow_tf32: no
use_8bit_adam: no
report_to: tensorboard
compute_gradient_conflict: no
compute_runtimes: yes
max_train_steps: 424
lr_warmup_steps: 0
adam_beta1: 0.9
adam_beta2: 0.999
adam_weight_decay: 0.01
adam_epsilon: 1e-08
max_grad_norm: 1
checkpointing_steps: 10000
noise_offset: 0
model-index: name: LeonardoBenitez/demo-vision-unlearning-munba
results: task: type: text-to-image
dataset: name: Forget set
type: inline-prompts
metrics: type: clip
value: 29.3575942332928
name: ForgetSet clip score of original model mean (~↑)
verified: no, type: clip
value: 2.362072773268124
name: ForgetSet clip score of original model std (~↓)
verified: no, type: clip
value: 18.763499865165123
name: ForgetSet clip score of learned model mean (~↑)
verified: no, type: clip
value: 2.365976164850448
name: ForgetSet clip score of learned model std (~↓)
verified: no, type: clip
value: 28.754149913787842
name: ForgetSet clip score of unlearned model mean (↓)
verified: no, type: clip
value: 2.52687691362335
name: ForgetSet clip score of unlearned model std (~↓)
verified: no, type: clip
value: -9.990650048622719
name: ForgetSet clip score difference between learned and unlearned mean (↑)
verified: no, type: clip
value: 3.1773814757306407
name: ForgetSet clip score difference between learned and unlearned std (~↓)
verified: no, type: clip
value: 0.6034443195049579
name: ForgetSet clip score difference between original and unlearned mean (↑)
verified: no, type: clip
value: 2.3739693718890127
name: ForgetSet clip score difference between original and unlearned std (~↓)
verified: no, type: clip
value: 10.594094368127676
name: ForgetSet clip score difference between original and learned mean (↓)
verified: no, type: clip
value: 2.8622713238514144
name: ForgetSet clip score difference between original and learned std (~↓)
verified: no, type: clip
value: 28.601373291015626
name: RetainSet clip score of original model mean (~↑)
verified: no, type: clip
value: 3.2160908507914225
name: RetainSet clip score of original model std (~↓)
verified: no, type: clip
value: 19.818220462799072
name: RetainSet clip score of learned model mean (~↓)
verified: no, type: clip
value: 2.8159758824147287
name: RetainSet clip score of learned model std (~↓)
verified: no, type: clip
value: 28.42326820373535
name: RetainSet clip score of unlearned model mean (↑)
verified: no, type: clip
value: 3.4659487211613795
name: RetainSet clip score of unlearned model std (~↓)
verified: no, type: clip
value: -8.605047740936278
name: RetainSet clip score difference between learned and unlearned mean (↓)
verified: no, type: clip
value: 3.878321726203634
name: RetainSet clip score difference between learned and unlearned std (~↓)
verified: no, type: clip
value: 0.17810508728027344
name: RetainSet clip score difference between original and unlearned mean (↓)
verified: no, type: clip
value: 2.9772598086693063
name: RetainSet clip score difference between original and unlearned std (~↓)
verified: no, type: clip
value: 8.783152828216553
name: RetainSet clip score difference between original and learned mean (↑)
verified: no, type: clip
value: 3.4170783629058983
name: RetainSet clip score difference between original and learned std (~↓)
verified: no, type: runtime
value: 3.67721261660258
name: Inference latency seconds mean (↓)
verified: no, type: runtime
value: 0.042937405905862674
name: Inference latency seconds std (~↓)
verified: no, task: type: text-to-image
dataset: name: assets/datasets/imagenette_splits/n03028079/train_forget (forget) and assets/datasets/imagenette_splits/n03028079/train_retain (retain) sets
type: forget-and-retain-together
metrics: type: runtime
value: 5.118733167648315
name: Runtime init seconds (~↓)
verified: no, type: runtime
value: 10.4545156955719
name: Runtime data loading seconds (~↓)
verified: no, type: runtime
value: 970.051931142807
name: Runtime training seconds (↓)
verified: no, type: runtime
value: 2514.1154568195343
name: Runtime eval seconds (~↓)
verified: no
Likes: 0 | All-time downloads: 0
Created: 2025-11-09 15:51:49