Spaces:
Running
Running
| project: | |
| project: starvector-im2svg | |
| use_wandb: false | |
| entity: abc | |
| copy_code: false | |
| model: | |
| max_length: 8192 | |
| model_name: null # in case of creating a new model, set this to None (null) | |
| starcoder_model_name: bigcode/starcoderbase-1b | |
| pretrained: true | |
| image_encoder_type: clip | |
| use_flash_attn: true | |
| adapter_norm: layer_norm | |
| init_type: glorot | |
| dropout: 0.1 | |
| task: im2svg | |
| transformer_layer_cls: None # fsdp specific | |
| 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 | |
| start_generation_at_step: 0 | |
| 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: ServiceNow/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: ServiceNow/svg-stack | |
| im_size: 224 | |
| num_samples: -1 | |
| transforms: false | |
| select_dataset_name: false | |
| generation: | |
| max_length: 8192 | |
| min_length: 10 | |
| num_beams: 3 | |
| num_captions: 1 | |
| num_generations_different_temp: 1.5 | |
| start_temperature: 0.5 | |
| repetition_penalty: 1.0 | |
| length_penalty: 1.0 | |
| temperature: 1.0 | |
| top_p: 0.9 | |
| use_nucleus_sampling: true | |
| im_size: 224 | |
| dpi: 2 | |
| scale: 300 | |
| num_samples_to_generate: -1 | |
| log_wandb_images: true | |
| start_generation_at_step: -1 | |
| 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 | |