rainbow/config.yaml

_wandb:
    value:
        cli_version: 0.24.2
        e:
            3idvx93n1kegu8y591ye7ef1ocd3ildl:
                args:
                    - --use_wandb
                    - --wandb_project_name=task1_all_kc
                    - --kc_to_que_path=data/XES3G5M/metadata/kc_questions_map.json
                    - --kc_emb_path=data/XES3G5M_embeddings/kc_emb.json
                    - --cluster_to_kc_path=data/XES3G5M/metadata/kc_clusters.json
                    - --cluster_to_que_path=data/XES3G5M/metadata/cluster_to_que_ids_map.json
                    - --pretrained_model_path=data/pretrained_kt_model.ckpt
                    - --dataloader_num_workers=8
                cpu_count: 252
                cpu_count_logical: 252
                cudaVersion: "12.8"
                disk:
                    /:
                        total: "1247017926656"
                        used: "907788537856"
                email: maxonorris@gmail.com
                executable: /mnt/pvc/Baselines/ExRec/.venv_task1/bin/python
                gpu: NVIDIA A100-SXM4-40GB
                gpu_count: 1
                gpu_nvidia:
                    - architecture: Ampere
                      cudaCores: 6912
                      memoryTotal: "42949672960"
                      name: NVIDIA A100-SXM4-40GB
                      uuid: GPU-9ae80bdd-e009-754a-7706-b530e2654492
                host: alphamcts-run-models-rainbow-mve-k598t-t45dv
                memory:
                    total: "962415210496"
                os: Linux-6.8.0-55-generic-x86_64-with-glibc2.31
                program: -m train_test.all_kc_train_rainbow
                python: CPython 3.10.13
                root: /mnt/pvc/Baselines/ExRec
                startedAt: "2026-02-12T13:50:37.537437Z"
                writerId: 3idvx93n1kegu8y591ye7ef1ocd3ildl
        m: []
        python_version: 3.10.13
        t:
            "1":
                - 1
                - 5
                - 53
            "2":
                - 1
                - 5
                - 53
            "3":
                - 2
                - 13
                - 15
                - 16
            "4": 3.10.13
            "5": 0.24.2
            "12": 0.24.2
            "13": linux-x86_64
action_size:
    value: 768
actor_lr:
    value: 5e-05
batch_size:
    value: 512
checkpoint_path:
    value: ./rainbow_saved_models/77b734bb-fe29-47e7-b671-8a40c8b5a79d
clip_loss_grad:
    value: false
cluster_to_kc_path:
    value: data/XES3G5M/metadata/kc_clusters.json
cluster_to_que_path:
    value: data/XES3G5M/metadata/cluster_to_que_ids_map.json
dataloader_num_workers:
    value: 8
discount_factor:
    value: 0.99
estimation_step:
    value: 1
hidden_size:
    value: 300
is_double:
    value: false
kc_emb_path:
    value: data/XES3G5M_embeddings/kc_emb.json
kc_emb_size:
    value: 768
kc_to_que_path:
    value: data/XES3G5M/metadata/kc_questions_map.json
log_dir:
    value: ./train_rainbow_logs
log_path:
    value: ./train_rainbow_logs/77b734bb-fe29-47e7-b671-8a40c8b5a79d
n_epoch:
    value: 100
num_atoms:
    value: 17
pretrained_model_path:
    value: data/pretrained_kt_model.ckpt
reward_normalization:
    value: false
sample_per_epoch:
    value: 20
save_dir:
    value: ./rainbow_saved_models
seed:
    value: 42
student_state_size:
    value: 300
target_update_freq:
    value: 0
test_batch_size:
    value: 2048
test_folds:
    value: "1"
test_init_seq_size:
    value: 100
test_last_n_steps:
    value: 10
test_log_wandb:
    value: false
test_max_steps:
    value: 10
test_n_episode:
    value: 2048
test_reward_func:
    value: step_by_step
test_reward_scale:
    value: 1000
train_batch_size:
    value: 512
train_folds:
    value: 2-3-4
train_init_seq_size:
    value: 100
train_last_n_steps:
    value: 10
train_max_steps:
    value: 10
train_max_steps_until_student_change:
    value: 10
train_n_episode:
    value: 512
train_replay_buffer_size:
    value: 100
train_reward_func:
    value: step_by_step
train_reward_scale:
    value: 1000
use_wandb:
    value: true
v_max:
    value: 1000
v_min:
    value: -1000
valid_folds:
    value: "0"
wandb_project_name:
    value: task1_all_kc
wandb_run_name:
    value: null