search

ai4co / rl4co

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

Update ppo.py #158

Closed bokveizen closed 5 months ago

bokveizen commented 5 months ago

Explicitly pass self.env into evaluate_action.

Otherwise, customed envs cannot work.