Upload figr8/art/config.yaml with huggingface_hub

figr8/art/config.yaml

@@ -1,66 +1,67 @@
+stage1_params:
+  name: VSQ
+  vector_decoder_model: mlp
+  quantized_dim: 512
+  codebook_size: 4096
+  image_loss: mse
+  vq_method: fsq
+  fsq_levels:
+  - 7
+  - 5
+  - 5
+  - 5
+  - 5
+  num_segments: 4
+  num_codes_per_shape: 2
+  alpha: 0.1
+  pred_color: false
+  lseg: 5.0
+  checkpoint_path: null
+model_params:
+  name: VQ_SVG_Stage2
+  max_seq_len: 562
+  dim: 512
+  ff_mult: 2
+  depth: 16
+  heads: 8
+  text_encoder_str: google/bert_uncased_L-12_H-512_A-8
+  use_alibi_positional_bias: false
 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
+  csv_path: null
+  vq_token_npy_path: null
+  train_batch_size: 16
+  val_batch_size: 16
+  num_workers: 32
+  width: 128
   max_text_token_length: 50
   min_context_length: 10
-  num_workers: 32
+  fraction_of_class_only_inputs: 0.1
+  fraction_of_blank_inputs: 0.1
+  fraction_of_iconshop_chatgpt_inputs: 0.25
+  fraction_of_full_description_inputs: 0.55
   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
+  grid_size: 256
 exp_params:
   lr: 0.0006
-  manual_seed: 42
-  post_process: false
+  weight_decay: 0.0001
   scheduler_gamma: 0.96
+  warmup_steps: 50000
   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
+  manual_seed: 42
+  post_process: false
 trainer_params:
-  accumulate_grad_batches: 2
   devices: -1
   max_epochs: 100
-
+  accumulate_grad_batches: 2
+logging_params:
+  entity: aiis-chair
+  project: grimoire-2
+  save_dir: null
+  name: ART_S4C2G256
+  version: 0
+  author: Marco
+_note: Correct config for the released figr8 ART (matches run that produced figr8/art/last.ckpt).
+  stage1_params here correctly rebuilds the figr8 VSQ.