example.json

{
    "max_num_stores_loads": 7,
    "max_num_loops": 7,
    "max_num_load_store_dim": 7,
    "num_tile_sizes": 7,
    "vect_size_limit": 2048,
    "order": [["TP"], ["V", "NT"]],
    "interchange_mode": "enumerate",
    "exploration": ["entropy"],
    "init_epsilon": 0.1,
    "new_architecture": false,
    "normalize_bounds": "max",
    "normalize_adv": "standard",
    "sparse_reward": true,
    "split_ops": false,
    "reuse_experience": "none",
    "activation": "relu",
    "benchmarks_folder_path": "",
    "bench_count": 20,
    "replay_count": 10,
    "nb_iterations": 10000,
    "ppo_epochs": 4,
    "ppo_batch_size": 4,
    "value_epochs": 32,
    "value_batch_size": 32,
    "value_coef": 0.5,
    "value_clip": false,
    "entropy_coef": 0.01,
    "lr": 0.001,
    "truncate": 5,
    "json_file": "data/nn/train_operations.json",
    "eval_json_file": "data/nn/eval_operations.json",
    "tags": ["nn"],
    "debug": false,
    "main_exec_data_file": "",
    "results_dir": "results"
}