Potpov
/

grimoire-checkpoints

+data_params:
+  csv_path: /raid/marco.cipriano/data/SVG/Grimoire/figr8/tokenized/VSQ_nseg=4_ncode=2_lseg=5.0_alpha=0.1None/split.csv
+  dataset: figr8
+  fraction_of_blank_inputs: 0.1
+  fraction_of_class_only_inputs: 0.1
+  fraction_of_full_description_inputs: 0.55
+  fraction_of_iconshop_chatgpt_inputs: 0.25
+  grid_size: 256
+  max_text_token_length: 50
+  min_context_length: 10
+  num_workers: 32
+  shuffle_vq_order: false
+  train_batch_size: 16
+  use_pre_computed_text_tokens_only: false
+  val_batch_size: 16
+  vq_token_npy_path: /raid/marco.cipriano/data/SVG/Grimoire/figr8/tokenized/VSQ_nseg=4_ncode=2_lseg=5.0_alpha=0.1None/tokenized.npy
+  width: 128
+exp_params:
+  lr: 0.0006
+  manual_seed: 42
+  post_process: false
+  scheduler_gamma: 0.96
+  train_log_interval: 0.25
+  val_log_interval: 0.5
+  warmup_steps: 50000
+  weight_decay: 0.0001
+logging_params:
+  author: Marco
+  entity: aiis-chair
+  name: ART_nseg=4_ncode=2_lseg=5_alpha01_256gri
+  project: thesis
+  save_dir: /raid/marco.cipriano/results/svg/Grimoire/ART/figr8/nseg=4_ncode=2_lseg=5_alpha01_256grid
+  version: 0
+model_params:
+  depth: 16
+  dim: 512
+  ff_mult: 2
+  heads: 8
+  max_seq_len: 562
+  name: VQ_SVG_Stage2
+  text_encoder_str: google/bert_uncased_L-12_H-512_A-8
+  use_alibi_positional_bias: false
+stage1_params:
+  alpha: 0.1
+  checkpoint_path: /raid/marco.cipriano/results/svg/Grimoire/VSQ/figr8/VSQ_nseg=4_ncode=2_lseg=5.0_alpha=0.1None/checkpoints/last.ckpt
+  codebook_size: 4096
+  fsq_levels:
+  - 7
+  - 5
+  - 5
+  - 5
+  - 5
+  image_loss: mse
+  lseg: 5.0
+  name: VSQ
+  num_codes_per_shape: 1
+  num_segments: 4
+  pred_color: false
+  quantized_dim: 512
+  vector_decoder_model: mlp
+  vq_method: fsq
+trainer_params:
+  accumulate_grad_batches: 2
+  devices: -1
+  max_epochs: 100