search

eleurent / rl-agents

Implementations of Reinforcement Learning and Planning algorithms
MIT License
553 stars 149 forks source link

hi, where is the code of computing social-attention and the code of updating the policy ? thanks. #102

Closed siatjia closed 11 months ago

siatjia commented 11 months ago

python experiments.py evaluate configs/IntersectionEnv/env.json \ configs/IntersectionEnv/agents/DQNAgent/ego_attention_2h.json \ --train --episodes=4000 --name-from-config

when i debug, i can not find it.

eleurent commented 11 months ago

Social attention: the architecture is in https://github.com/eleurent/rl-agents/blob/63d3734123fffcfcc84d9fc84195256a94d74638/rl_agents/agents/common/models.py#L157

Policy update: https://github.com/eleurent/rl-agents/blob/63d3734123fffcfcc84d9fc84195256a94d74638/rl_agents/agents/deep_q_network/pytorch.py#L14

siatjia commented 11 months ago

Thanks. good job.