See files under walk_the_blocks/BlockWorldRoboticAgent/srcs/
-
learn_by_ppo.py
run this file for training, you can change the schedule mechanism in the functionppo_update()
, these are the options:- do imitation every 50
- do imitation based on rules
- imitation 1 epoch and then RL 1 epoch
example:
python learn_by_ppo.py -lr 0.0001 -max_epochs 2 -entropy_coef 0.05
-
policy_model.py
the network achitecture and loss functions:- PPO Loss
- Supervised Loss
- Advantage Actor-Critic Loss
For the usage of the Block-world environment, please refer to https://github.com/clic-lab/blocks
- S-REIN *