↓ 2 callersMethodMultiHeadsAttModel input: [batch, agent, dim] feature [batch, agent, nei, agent] adjacency input:[bacth,agent,128] output:
models/colight_agent.py:96
↓ 1 callersMethodMLP Currently, the MLP layer -input: [batch,#agents,dim] -outpout: [batch,#agents,dim]
models/colight_agent.py:78
↓ 1 callersMethod__init__(self, dic_agent_conf=None, dic_traffic_env_conf=None, dic_path=None, cnt_round=None,
interse
models/colight_agent.py:25