trap/EXPERIMENTS/config.json

{
    "batch_size": 512,
    "grad_clip": 1.0,
    "learning_rate_style": "exp",
    "learning_rate": 0.01, 
    "min_learning_rate": 1e-05,
    "learning_decay_rate": 0.9999,
    "prediction_horizon": 30,
    "minimum_history_length": 5,
    "maximum_history_length": 50,
    "map_encoder": {
        "PEDESTRIAN": {
            "heading_state_index": [2, 3],
            "patch_size": [
                50,
                10,
                50,
                90
            ],
            "map_channels": 3,
            "hidden_channels": [
                10,
                20,
                10,
                1
            ],
            "output_size": 32,
            "masks": [
                5,
                5,
                5,
                5
            ],
            "strides": [
                1,
                1,
                1,
                1
            ],
            "dropout": 0.5
        }
    },
    "k": 1,
    "k_eval": 25,
    "kl_min": 0.07,
    "kl_weight": 100.0,
    "kl_weight_start": 0,
    "kl_decay_rate": 0.99995,
    "kl_crossover": 400,
    "kl_sigmoid_divisor": 4,
    "rnn_kwargs": {
        "dropout_keep_prob": 0.75
    },
    "MLP_dropout_keep_prob": 0.9,
    "enc_rnn_dim_edge": 32,
    "enc_rnn_dim_edge_influence": 32,
    "enc_rnn_dim_history": 32,
    "enc_rnn_dim_future": 32,
    "dec_rnn_dim": 128,
    "q_z_xy_MLP_dims": null,
    "p_z_x_MLP_dims": 32,
    "GMM_components": 1,
    "log_p_yt_xz_max": 6,
    "N": 1,
    "K": 25,
    "tau_init": 2.0,
    "tau_final": 0.05,
    "tau_decay_rate": 0.997,
    "use_z_logit_clipping": true,
    "z_logit_clip_start": 0.05,
    "z_logit_clip_final": 5.0,
    "z_logit_clip_crossover": 300,
    "z_logit_clip_divisor": 5,
    "dynamic": {
        "PEDESTRIAN": {
            "name": "SingleIntegrator",
            "distribution": true,
            "limits": {}
        }
    },
    "state": {
        "PEDESTRIAN": {
            "position": [
                "x",
                "y"
            ],
            "velocity": [
                "x",
                "y"
            ],
            "acceleration": [
                "x",
                "y"
            ]
        }
    },
    "pred_state": {
        "PEDESTRIAN": {
            "position": [
                "x",
                "y"
            ]
        }
    },
    "log_histograms": false,
    "dynamic_edges": "yes",
    "edge_state_combine_method": "sum",
    "edge_influence_combine_method": "attention",
    "edge_addition_filter": [
        0.25,
        0.5,
        0.75,
        1.0
    ],
    "edge_removal_filter": [
        1.0,
        0.0
    ],
    "offline_scene_graph": "yes",
    "incl_robot_node": false,
    "node_freq_mult_train": false,
    "node_freq_mult_eval": false,
    "scene_freq_mult_train": false,
    "scene_freq_mult_eval": false,
    "scene_freq_mult_viz": false,
    "edge_encoding": true,
    "use_map_encoding": true,
    "augment": false,
    "override_attention_radius": []
}