charaatasnim commited on
Commit
bdd49db
1 Parent(s): 4f960bb

Upload 2 files

Browse files
events.out.tfevents.1714466142.874760fc3fd2.128.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:276ac9313405cd9f7c0762cb59fb62a5ba5d3858cbc861f1a0bdf2444fb214f6
3
+ size 2416
hparams.yml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: null
7
+ center_crop: false
8
+ checkpointing_steps: 2
9
+ checkpoints_total_limit: 2
10
+ conditioning_dropout_prob: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: annyorange/colorized_people-dataset
14
+ edit_prompt_column: edit_prompt
15
+ edited_image_column: colorized_image
16
+ enable_xformers_memory_efficient_attention: false
17
+ gradient_accumulation_steps: 4
18
+ gradient_checkpointing: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ learning_rate: 1.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: 766
28
+ max_train_steps: 5
29
+ mixed_precision: null
30
+ non_ema_revision: null
31
+ num_train_epochs: 1
32
+ num_validation_images: 1
33
+ original_image_column: original_image
34
+ output_dir: instruct-pix2pix-model
35
+ pretrained_model_name_or_path: timbrooks/instruct-pix2pix
36
+ push_to_hub: false
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 256
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: 42
44
+ train_batch_size: 2
45
+ train_data_dir: null
46
+ use_8bit_adam: false
47
+ use_ema: false
48
+ val_batch_size: 4
49
+ val_gt_url: null
50
+ val_image_url: null
51
+ validation_epochs: 1
52
+ validation_prompt: null