Skip to content

TianxiangZhao/ImitationNoisyDemon

Repository files navigation

ImitationNoisyDemon

Dependencies

  • pytorch
  • imitation
  • imblearn
  • stable_baselines3==1.5.0
  • gym-minigrid==1.0.3

Usage Instructions

  1. Demonstration generation

Configurations are set inside each python file.

> python train_expert.py

> python collect_traj.py

  1. Test model

Example 1:

> python imit.py

Example 2:

> python main_Fourrooms.py

About

Codes for Skill Disentanglement for Imitation Learning from Suboptimal Demonstrations, KDD23

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages