epicgamer17 / rl-research

some rl research and paper recreations
3 stars 0 forks source link

make ppo actor and critic into one network (that just has to heads that branch at the input) #46

Closed epicgamer17 closed 8 months ago

epicgamer17 commented 8 months ago

just made a network that has the other two networks as attributes and can call them using model.critic or model.actor