arpachat commited on
Commit
5614260
·
1 Parent(s): 5742b8d

End of training

Browse files
README.md CHANGED
@@ -37,9 +37,9 @@ image.save("my_image.png")
37
 
38
  These are the key hyperparameters used during training:
39
 
40
- * Epochs: 16
41
  * Learning rate: 1e-05
42
- * Batch size: 1
43
  * Gradient accumulation steps: 2
44
  * Image resolution: 512
45
  * Mixed-precision: fp16
 
37
 
38
  These are the key hyperparameters used during training:
39
 
40
+ * Epochs: 58
41
  * Learning rate: 1e-05
42
+ * Batch size: 4
43
  * Gradient accumulation steps: 2
44
  * Image resolution: 512
45
  * Mixed-precision: fp16
logs/text2image-fine-tune/1702515829.435362/events.out.tfevents.1702515829.69a9ca71ebd3.26389.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0cd96b91608c5ba755930f71e851dac683a1e9875861f248bf6fdab07e864c49
3
+ size 2333
logs/text2image-fine-tune/1702515829.4372208/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 10000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: jwl25b/final_project_dataset
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 2
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: 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
19
+ image_column: image
20
+ input_perturbation: 0
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: null
28
+ max_train_steps: 400
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 400
33
+ output_dir: output-fashion-400
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: OFA-Sys/small-stable-diffusion-v0
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ tracker_project_name: text2image-fine-tune
46
+ train_batch_size: 64
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ use_ema: true
50
+ validation_epochs: 5
51
+ variant: null
logs/text2image-fine-tune/1702515865.5909758/events.out.tfevents.1702515865.69a9ca71ebd3.26560.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81eb6394e9ad1fbd7bb83423b2923cea6b94bee68f3a55f33c369eb11b9fd62f
3
+ size 2333
logs/text2image-fine-tune/1702515865.5927641/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 10000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: jwl25b/final_project_dataset
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 2
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: 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
19
+ image_column: image
20
+ input_perturbation: 0
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: null
28
+ max_train_steps: 400
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 200
33
+ output_dir: output-fashion-400
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: OFA-Sys/small-stable-diffusion-v0
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ tracker_project_name: text2image-fine-tune
46
+ train_batch_size: 16
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ use_ema: true
50
+ validation_epochs: 5
51
+ variant: null
logs/text2image-fine-tune/1702515908.8860111/events.out.tfevents.1702515908.69a9ca71ebd3.26762.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9512fa8f9d8ccd5a8e529299ac8513d88d5e71d07a8d5257ebea4680b446d1e8
3
+ size 2333
logs/text2image-fine-tune/1702515908.8878531/hparams.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 10000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: jwl25b/final_project_dataset
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 2
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: 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
19
+ image_column: image
20
+ input_perturbation: 0
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: null
28
+ max_train_steps: 400
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 58
33
+ output_dir: output-fashion-400
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: OFA-Sys/small-stable-diffusion-v0
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: null
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ tracker_project_name: text2image-fine-tune
46
+ train_batch_size: 4
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ use_ema: true
50
+ validation_epochs: 5
51
+ variant: null
logs/text2image-fine-tune/events.out.tfevents.1702514554.69a9ca71ebd3.20961.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b330748267967e2e6c9a5a3da16d8cc8939cf49f91bd39d46faa8db97b7eb1f8
3
- size 1148390
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d5e044042d75e74849db5a6eeb7fc526d5a76b449adc064612c5fa7ade791fc
3
+ size 1152166
logs/text2image-fine-tune/events.out.tfevents.1702515829.69a9ca71ebd3.26389.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11b5c9ec346568c39a8f6fb8974309b654cfb246e46de29a09c059f8240be338
3
+ size 88
logs/text2image-fine-tune/events.out.tfevents.1702515865.69a9ca71ebd3.26560.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71acf8bfbaa915df11d694246a57656682200f65c42f97e016b8bb67138af2cf
3
+ size 136
logs/text2image-fine-tune/events.out.tfevents.1702515908.69a9ca71ebd3.26762.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2413135b34eae42f17f12b42bb8dbcb530fcfbe17136d8e2f7c40e21e85c28fb
3
+ size 3538211
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f57566604c90cb4696dcf46345e89d517b91b96bfb01a28edfabb18d6e3ced7
3
  size 2317593424
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0312d583e2beb69cb7f2e71a888a579ef2c40d46dcb8a0d9481364d4edffb82e
3
  size 2317593424
val_imgs_grid.png CHANGED