Hugging Face's logo

kfirgold99
/
Piece-it-Together

Model card Files
xet
Community
1
eladrich commited on
Commit
ceb4b26
·
verified ·
1 Parent(s): c39213f

Upload 6 files

Browse files

Added PiT ckpts

Files changed (6) hide show
  1. models/characters_ckpt/cfg.yaml +28 -0
  2. models/characters_ckpt/prior.ckpt +3 -0
  3. models/plush_ckpt/cfg.yaml +28 -0
  4. models/plush_ckpt/prior.ckpt +3 -0
  5. models/products_ckpt/cfg.yaml +28 -0
  6. models/products_ckpt/prior.ckpt +3 -0
models/characters_ckpt/cfg.yaml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0
5
+ checkpointing_steps: 5000
6
+ dataset_path: datasets/generated/characters_train
7
+ device: cuda:0
8
+ gradient_accumulation_steps: 1
9
+ hidden_dim: 2048
10
+ img_size: 1024
11
+ log_image_frequency: 500
12
+ log_validation: 5000
13
+ lr: 1.0e-05
14
+ max_crops: 7
15
+ max_grad_norm: 1.0
16
+ max_train_steps: 1000000
17
+ mixed_precision: fp16
18
+ n_val_images: 10
19
+ num_attention_heads: 32
20
+ num_layers: 4
21
+ num_workers: 8
22
+ report_to: wandb
23
+ resume_from_step: null
24
+ seed: null
25
+ sketch_prob: 0.5
26
+ train_batch_size: 64
27
+ use_ref: true
28
+ val_dataset_path: datasets/generated/characters_val
models/characters_ckpt/prior.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab7e886db38e2d32789579f39c39404727bdadf6981ff0b3c2bf39c17c2f2430
3
+ size 1079507976
models/plush_ckpt/cfg.yaml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0
5
+ checkpointing_steps: 5000
6
+ dataset_path: datasets/generated/plush_train_set
7
+ device: cuda:0
8
+ gradient_accumulation_steps: 1
9
+ hidden_dim: 2048
10
+ img_size: 1024
11
+ log_image_frequency: 500
12
+ log_validation: 5000
13
+ lr: 1.0e-05
14
+ max_crops: 7
15
+ max_grad_norm: 1.0
16
+ max_train_steps: 1000000
17
+ mixed_precision: fp16
18
+ n_val_images: 10
19
+ num_attention_heads: 32
20
+ num_layers: 4
21
+ num_workers: 8
22
+ report_to: wandb
23
+ resume_from_step: null
24
+ seed: null
25
+ sketch_prob: 0.0
26
+ train_batch_size: 64
27
+ use_ref: false
28
+ val_dataset_path: datasets/generated/plush_val_set
models/plush_ckpt/prior.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63e17c42d6d8ff1add2a2ff021f89214f56230be3b6fcee74d1f94e2139aa9bf
3
+ size 1079507976
models/products_ckpt/cfg.yaml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0
5
+ checkpointing_steps: 5000
6
+ dataset_path: datasets/generated/product_parts_train
7
+ device: cuda:0
8
+ gradient_accumulation_steps: 1
9
+ hidden_dim: 2048
10
+ img_size: 1024
11
+ log_image_frequency: 500
12
+ log_validation: 5000
13
+ lr: 1.0e-05
14
+ max_crops: 7
15
+ max_grad_norm: 1.0
16
+ max_train_steps: 1000000
17
+ mixed_precision: fp16
18
+ n_val_images: 10
19
+ num_attention_heads: 32
20
+ num_layers: 4
21
+ num_workers: 8
22
+ report_to: wandb
23
+ resume_from_step: null
24
+ seed: null
25
+ sketch_prob: 0.0
26
+ train_batch_size: 64
27
+ use_ref: false
28
+ val_dataset_path: datasets/generated/product_parts_val
models/products_ckpt/prior.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:844251bad334dc648169e920e51a06a0b17f336e76ddbc30e7e1409504563c7c
3
+ size 1079507976