Add v7.2-mini, delete old

v6.0/checkpoint_100000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_100000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_100000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_100000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_100000/state_dict.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:db72b9b9e1769570b0169dff54a6695e37f1d24d7eddd98e4fff401e7c9ecf6e
-size 108624145

v6.0/checkpoint_1000000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_1000000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_1000000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1000000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1100000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_1100000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_1100000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1100000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1200000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_1200000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_1200000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1200000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1300000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_1300000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_1300000/nsr.yaml

@@ -1,20 +0,0 @@
-encoder_max_n_variables: 4
-size: 512
-pre_encoder_input_type: ieee-754
-pre_encoder_support_nan: false
-encoder: SetTransformer
-encoder_kwargs:
-  hidden_size: 512
-  n_enc_isab: 5
-  n_dec_sab: 2
-  n_induce: 64
-  n_heads: 8
-  layer_norm: false
-  n_seeds: 64
-decoder_n_heads: 8
-decoder_ff_size: 512
-decoder_dropout: 0.1
-decoder_n_layers: 5
-learnable_positional_embeddings: false
-max_input_length: null
-expression_space: ./expression_space.yaml

v6.0/checkpoint_1300000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1300000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1300000/state_dict.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6cb6bf0f8bf840bde7a98bc641f2a434de90c5f48ddf02907b94a24fe9a4b57e
-size 108624145

v6.0/checkpoint_1300000/train.yaml

@@ -1,22 +0,0 @@
-model: ./nsr.yaml
-optimizer:
-  name: AdamW
-  kwargs:
-    lr: 1
-    weight_decay: 0.01
-    amsgrad: true
-lr_scheduler:
-  name: WarmupLinearAnnealing
-  kwargs:
-    min_lr: 0
-    max_lr: 1e-4
-    warmup_steps: 10000
-    total_steps: 1500000
-batch_size: 128
-train_dataset: ./dataset_train.yaml
-val_dataset: ./dataset_val.yaml
-val_batch_size: 128
-val_size: 100000
-numeric_prediction_loss_weight: 0
-steps: 1500000
-device: cuda

v6.0/checkpoint_1400000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_1400000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_1400000/nsr.yaml

@@ -1,20 +0,0 @@
-encoder_max_n_variables: 4
-size: 512
-pre_encoder_input_type: ieee-754
-pre_encoder_support_nan: false
-encoder: SetTransformer
-encoder_kwargs:
-  hidden_size: 512
-  n_enc_isab: 5
-  n_dec_sab: 2
-  n_induce: 64
-  n_heads: 8
-  layer_norm: false
-  n_seeds: 64
-decoder_n_heads: 8
-decoder_ff_size: 512
-decoder_dropout: 0.1
-decoder_n_layers: 5
-learnable_positional_embeddings: false
-max_input_length: null
-expression_space: ./expression_space.yaml

v6.0/checkpoint_1400000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1400000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1400000/state_dict.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7b07c0285399d2b9b3f0a167cd1680ff35e60287be6ce0a5ece2a0903717c7ff
-size 108624145

v6.0/checkpoint_1400000/train.yaml

@@ -1,22 +0,0 @@
-model: ./nsr.yaml
-optimizer:
-  name: AdamW
-  kwargs:
-    lr: 1
-    weight_decay: 0.01
-    amsgrad: true
-lr_scheduler:
-  name: WarmupLinearAnnealing
-  kwargs:
-    min_lr: 0
-    max_lr: 1e-4
-    warmup_steps: 10000
-    total_steps: 1500000
-batch_size: 128
-train_dataset: ./dataset_train.yaml
-val_dataset: ./dataset_val.yaml
-val_batch_size: 128
-val_size: 100000
-numeric_prediction_loss_weight: 0
-steps: 1500000
-device: cuda

v6.0/checkpoint_1500000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_1500000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_1500000/nsr.yaml

@@ -1,20 +0,0 @@
-encoder_max_n_variables: 4
-size: 512
-pre_encoder_input_type: ieee-754
-pre_encoder_support_nan: false
-encoder: SetTransformer
-encoder_kwargs:
-  hidden_size: 512
-  n_enc_isab: 5
-  n_dec_sab: 2
-  n_induce: 64
-  n_heads: 8
-  layer_norm: false
-  n_seeds: 64
-decoder_n_heads: 8
-decoder_ff_size: 512
-decoder_dropout: 0.1
-decoder_n_layers: 5
-learnable_positional_embeddings: false
-max_input_length: null
-expression_space: ./expression_space.yaml

v6.0/checkpoint_1500000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1500000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_1500000/state_dict.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:de20334451ffda2d7fe4d9b3a76a5be622304733911a937a2db255a808c19073
-size 108624145

v6.0/checkpoint_1500000/train.yaml

@@ -1,22 +0,0 @@
-model: ./nsr.yaml
-optimizer:
-  name: AdamW
-  kwargs:
-    lr: 1
-    weight_decay: 0.01
-    amsgrad: true
-lr_scheduler:
-  name: WarmupLinearAnnealing
-  kwargs:
-    min_lr: 0
-    max_lr: 1e-4
-    warmup_steps: 10000
-    total_steps: 1500000
-batch_size: 128
-train_dataset: ./dataset_train.yaml
-val_dataset: ./dataset_val.yaml
-val_batch_size: 128
-val_size: 100000
-numeric_prediction_loss_weight: 0
-steps: 1500000
-device: cuda

v6.0/checkpoint_200000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_200000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_200000/nsr.yaml

@@ -1,20 +0,0 @@
-encoder_max_n_variables: 4
-size: 512
-pre_encoder_input_type: ieee-754
-pre_encoder_support_nan: false
-encoder: SetTransformer
-encoder_kwargs:
-  hidden_size: 512
-  n_enc_isab: 5
-  n_dec_sab: 2
-  n_induce: 64
-  n_heads: 8
-  layer_norm: false
-  n_seeds: 64
-decoder_n_heads: 8
-decoder_ff_size: 512
-decoder_dropout: 0.1
-decoder_n_layers: 5
-learnable_positional_embeddings: false
-max_input_length: null
-expression_space: ./expression_space.yaml

v6.0/checkpoint_200000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_200000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_200000/state_dict.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7961ec463097ff830dbff9a1e20b18bd38231376245c2dc5a8ef3e72fa81a5f1
-size 108624145

v6.0/checkpoint_200000/train.yaml

@@ -1,22 +0,0 @@
-model: ./nsr.yaml
-optimizer:
-  name: AdamW
-  kwargs:
-    lr: 1
-    weight_decay: 0.01
-    amsgrad: true
-lr_scheduler:
-  name: WarmupLinearAnnealing
-  kwargs:
-    min_lr: 0
-    max_lr: 1e-4
-    warmup_steps: 10000
-    total_steps: 1500000
-batch_size: 128
-train_dataset: ./dataset_train.yaml
-val_dataset: ./dataset_val.yaml
-val_batch_size: 128
-val_size: 100000
-numeric_prediction_loss_weight: 0
-steps: 1500000
-device: cuda

v6.0/checkpoint_300000/dataset_train.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_train.yaml
-padding: zero

v6.0/checkpoint_300000/dataset_val.yaml

@@ -1,2 +0,0 @@
-skeleton_pool: ./skeleton_pool_val.yaml
-padding: zero

v6.0/checkpoint_300000/nsr.yaml

@@ -1,20 +0,0 @@
-encoder_max_n_variables: 4
-size: 512
-pre_encoder_input_type: ieee-754
-pre_encoder_support_nan: false
-encoder: SetTransformer
-encoder_kwargs:
-  hidden_size: 512
-  n_enc_isab: 5
-  n_dec_sab: 2
-  n_induce: 64
-  n_heads: 8
-  layer_norm: false
-  n_seeds: 64
-decoder_n_heads: 8
-decoder_ff_size: 512
-decoder_dropout: 0.1
-decoder_n_layers: 5
-learnable_positional_embeddings: false
-max_input_length: null
-expression_space: ./expression_space.yaml

v6.0/checkpoint_300000/skeleton_pool_train.yaml

@@ -1,31 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/v6.0/skeleton_pool_val/'
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512

v6.0/checkpoint_300000/skeleton_pool_val.yaml

@@ -1,30 +0,0 @@
-expression_space: ./expression_space.yaml
-holdout_pools:
-- '{{ROOT}}/data/ansr-data/test_set/soose_nc/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/feynman/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/nguyen/skeleton_pool/'
-- '{{ROOT}}/data/ansr-data/test_set/pool_15/skeleton_pool/'
-sample_strategy:
-  n_operator_distribution: length_proportional
-  min_operators: 0
-  max_operators: 10
-  power: 1
-  max_length: 21
-  max_tries: 1
-  independent_dimensions: true
-allow_nan: false
-simplify: true
-literal_prior: uniform
-literal_prior_kwargs:
-  low: -5
-  high: 5
-support_prior: uniform_intervals
-support_prior_kwargs:
-  low: -10
-  high: 10
-n_support_prior: uniform
-n_support_prior_kwargs:
-  low: 16
-  high: 512
-  min_value: 16
-  max_value: 512