_wandb:
    value:
        cli_version: 0.30.0
        e:
            kp4yvqefp3ui3qebj702nb557950820s:
                args:
                    - -of
                    - /config/run_a_v0_main_fixed.json
                    - -l
                    - /scratch
                    - -n
                    - c2v-A-v0main-fixed
                cpu_count: 128
                cpu_count_logical: 256
                cudaVersion: "13.4"
                disk:
                    /:
                        total: "1599952912384"
                        used: "523544772608"
                email: mkq5008@gmail.com
                executable: /opt/conda/bin/python
                gpu: NVIDIA A100-SXM4-80GB
                gpu_count: 1
                gpu_nvidia:
                    - architecture: Ampere
                      cudaCores: 6912
                      memoryTotal: "85899345920"
                      name: NVIDIA A100-SXM4-80GB
                      uuid: GPU-e477ea48-7ddf-dba1-d7fd-e6cb06b29150
                host: mq-train-c2v-a-v0main-fixed-mxk6q
                memory:
                    total: "1081702121472"
                os: Linux-6.8.0-146-generic-x86_64-with-glibc2.36
                program: -m spanet.train
                python: CPython 3.12.10
                root: /scratch/c2v-A-v0main-fixed
                startedAt: "2026-10-10T04:23:15.275114Z"
                writerId: kp4yvqefp3ui3qebj702nb557950820s
        m:
            - "1": trainer/global_step
              "6":
                - 3
              "7": []
            - "2": '*'
              "5": 1
              "6":
                - 1
              "7": []
        python_version: 3.12.10
        t:
            "1":
                - 1
                - 5
                - 9
                - 53
                - 103
            "2":
                - 1
                - 5
                - 9
                - 53
                - 103
            "3":
                - 7
                - 13
                - 15
                - 41
                - 66
            "4": 3.12.10
            "5": 0.30.0
            "12": 0.30.0
            "13": linux-x86_64
assignment_loss_scale:
    value: 1
balance_classifications:
    value: false
balance_jets:
    value: 0
balance_losses:
    value: true
balance_particles:
    value: 0
batch_size:
    value: 1024
classification_loss_scale:
    value: 0
combinatorial_scale:
    value: 0
combine_pair_loss:
    value: min
dataset_limit:
    value: 1
dataset_randomization:
    value: 1
detection_loss_scale:
    value: 1
dropout:
    value: 0.2
epochs:
    value: 500
event_info_file:
    value: /config/tt_hadronic_notag_fixed.yaml
focal_gamma:
    value: 0
gradient_clip:
    value: 10
hidden_dim:
    value: 128
initial_embedding_dim:
    value: 128
initial_embedding_skip_connections:
    value: 1
kl_loss_scale:
    value: 0
l2_penalty:
    value: 0.0002
learning_rate:
    value: 0.0004
learning_rate_cycles:
    value: 1
learning_rate_warmup_epochs:
    value: 1
limit_to_num_jets:
    value: 0
limit_val_batches:
    value: 1
linear_activation:
    value: gelu
linear_block_type:
    value: GRU
linear_prelu_activation:
    value: true
mask_sequence_vectors:
    value: 1
masking:
    value: Filling
mass_classes:
    value: []
moe_loss_scale:
    value: 0.01
normalization:
    value: LayerNorm
normalize_features:
    value: 1
num_attention_heads:
    value: 8
num_branch_embedding_layers:
    value: 3
num_branch_encoder_layers:
    value: 3
num_classification_layers:
    value: 3
num_dataloader_workers:
    value: 0
num_detector_layers:
    value: 2
num_embedding_layers:
    value: 10
num_encoder_layers:
    value: 4
num_experts:
    value: 8
num_experts_per_tok:
    value: 2
num_gpu:
    value: 1
num_jet_embedding_layers:
    value: 0
num_jet_encoder_layers:
    value: 2
num_pairwise_features:
    value: 4
num_regression_layers:
    value: 3
optimizer:
    value: AdamW
pairwise_block_embeddings:
    value: false
pairwise_cross_type:
    value: true
pairwise_embedding_dim:
    value: 8
pairwise_eq2to2:
    value: false
pairwise_feature_set:
    value: ""
pairwise_input_source:
    value: ""
partial_events:
    value: 1
position_embedding_dim:
    value: 128
regression_loss_scale:
    value: 0
run_dir:
    value: /scratch/c2v-A-v0main-fixed
skip_connections:
    value: 1
spanet_ref:
    value: aaa5635cd51bdd30926dbc91234271398cf7412a
spatop_git_sha:
    value: 036d6f0
split_symmetric_attention:
    value: 1
testing_file:
    value: ""
train_validation_split:
    value: 1
training_file:
    value: /traindatavol/collide2v/tt_hadronic_1M_v0_main/tt_hadronic_training.h5
transformer_activation:
    value: gelu
transformer_dim:
    value: 128
transformer_dim_scale:
    value: 2
transformer_type:
    value: Gated
trial_output_dir:
    value: ./test_output
trial_time:
    value: ""
usable_gpus:
    value: ""
use_mass_classification:
    value: false
use_moe:
    value: false
use_pairwise_interactions:
    value: false
validation_file:
    value: /traindatavol/collide2v/tt_hadronic_1M_v0_main/tt_hadronic_validation.h5
verbose_output:
    value: false