- pytorch
- imitation
- imblearn
- stable_baselines3==1.5.0
- gym-minigrid==1.0.3
- Demonstration generation
Configurations are set inside each python file.
> python train_expert.py
> python collect_traj.py
- Test model
Example 1:
> python imit.py
Example 2:
> python main_Fourrooms.py