search

ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
https://rl4.co
MIT License
455 stars 84 forks source link

Update ppo.py #158

Closed bokveizen closed 7 months ago

bokveizen commented 7 months ago

Explicitly pass self.env into evaluate_action.

Otherwise, customed envs cannot work.