|
project: |
|
project: starvector-1b-im2svg |
|
use_wandb: false |
|
entity: abc |
|
copy_code: false |
|
model: |
|
max_length: 8192 |
|
model_name: null |
|
starcoder_model_name: bigcode/starcoderbase-1b |
|
pretrained: true |
|
image_encoder_type: clip |
|
use_flash_attn: true |
|
adapter_norm: batch_norm |
|
init_type: glorot |
|
dropout: 0.1 |
|
task: im2svg |
|
transformer_layer_cls: null |
|
use_cache: false |
|
training: |
|
save_model_epochs: 1 |
|
checkpointing_steps: 500 |
|
checkpoints_total_limit: 3 |
|
model_precision: bf16 |
|
resume_from_checkpoint: false |
|
continue_training: false |
|
n_epochs: 4 |
|
lr: 0.00001 |
|
gradient_accumulation_steps: 4 |
|
lr_scheduler: cosine |
|
lr_warmup_steps: 10 |
|
adam_beta1: 0.95 |
|
adam_beta2: 0.999 |
|
adam_weight_decay: 1.0e-06 |
|
adam_epsilon: 1e-08 |
|
optimizer: adamw |
|
train_image_encoder: true |
|
train_LLM: true |
|
use_gradient_checkpointing: false |
|
fsdp: |
|
enable: false |
|
data: |
|
num_workers: 4 |
|
train: |
|
batch_size: 2 |
|
target: starvector.data.stacksvg.SVGStackDataset |
|
params: |
|
split: train |
|
dataset_name: starvector/svg-stack |
|
im_size: 224 |
|
num_samples: -1 |
|
transforms: false |
|
select_dataset_name: false |
|
test: |
|
batch_size: 2 |
|
target: starvector.data.stacksvg.SVGStackDataset |
|
params: |
|
split: test |
|
dataset_name: starvector/svg-stack |
|
im_size: 224 |
|
num_samples: -1 |
|
transforms: false |
|
select_dataset_name: false |
|
generation: |
|
max_length: 8192 |
|
min_length: 10 |
|
num_beams: 3 |
|
temperature: 1.0 |
|
num_captions: 1 |
|
repetition_penalty: 1.0 |
|
length_penalty: 0.5 |
|
top_p: 0.95 |
|
use_nucleus_sampling: true |
|
im_size: 224 |
|
dpi: 2 |
|
scale: 300 |
|
metrics: |
|
L2: true |
|
Masked-L2: false |
|
LPIPS: true |
|
SSIM: true |
|
FID: false |
|
FID_clip: false |
|
CLIPScore: false |
|
CountTokenLength: true |
|
ratio_post_processed: false |
|
ratio_non_compiling: false |
|
DinoScore: true |
|
|
