pkm_config.json

{
    "mem_implementation": "pq_fast",
    "mem_grouped_conv": 0,
    "mem_values_optimizer": "",
    "mem_sparse": 0,
    "mem_input2d": 0,
    "mem_k_dim": 256,
    "mem_v_dim": -1,
    "mem_heads": 4,
    "mem_knn": 32,
    "mem_share_values": 0,
    "mem_shuffle_indices": 0,
    "mem_shuffle_query": 0,
    "mem_modulo_size": -1,
    "mem_keys_type": "uniform",
    "mem_n_keys": 512,
    "mem_keys_normalized_init": 0,
    "mem_keys_learn": 1,
    "mem_use_different_keys": 1,
    "mem_query_detach_input": 0,
    "mem_query_layer_sizes": "0,0",
    "mem_query_kernel_sizes": "",
    "mem_query_bias": 1,
    "mem_query_batchnorm": 0,
    "mem_query_net_learn": 1,
    "mem_query_residual": 0,
    "mem_multi_query_net": 0,
    "mem_value_zero_init": 0,
    "mem_normalize_query": 1,
    "mem_temperature": 1,
    "mem_score_softmax": 1,
    "mem_score_subtract": "",
    "mem_score_normalize": 0,
    "mem_input_dropout": 0,
    "mem_query_dropout": 0,
    "mem_value_dropout": 0
}