adding function to load experts and combine them into single transition list

2021-10-15 02:47:52 -07:00
parent ffb16cfc31
commit 62bfe797a4
4 changed files with 28 additions and 2 deletions
--- a/scratch/etienne/intersimple/gail_options_image.py
+++ b/scratch/etienne/intersimple/gail_options_image.py
@@ -303,6 +303,8 @@ if __name__ == '__main__':
 
    with open("data/NormalizedIntersimpleExpertMu.001_NRasterizedAgent51w36h36mppx2.pkl", "rb") as f:
        trajectories = pickle.load(f)
+    import pdb 
+    pdb.set_trace()
    transitions = rollout.flatten_trajectories(trajectories)
    generator = train(transitions)