diff --git a/generate_demos.sh b/generate_demos.sh deleted file mode 100755 index 0e711b4..0000000 --- a/generate_demos.sh +++ /dev/null @@ -1,10 +0,0 @@ -#DEFAULT PARAMETERS: -# locs:list=None, (default to all locations) -# tracks:list=None, (default to all tracks) -# env_class:str='NRasterizedIncrementingAgent', -# env_args:dict={width:36,height:36,m_per_px:2}, -# expert_class:str='NRasterizedRouteIncrementingAgent', -# expert_args:dict={mu:0.001}): - -# python -m src.data.expert --locs='[DR_USA_Roundabout_FT]' --tracks='[0]' -python -m src.data.expert --locs='[DR_USA_Roundabout_FT]' --tracks='[0]' \ No newline at end of file diff --git a/intersimple-expert-data-setobs2.pt b/intersimple-expert-data-setobs2.pt deleted file mode 100644 index 62323d4..0000000 Binary files a/intersimple-expert-data-setobs2.pt and /dev/null differ diff --git a/intersimple-expert-rollout-setobs2.py b/intersimple-expert-rollout-setobs2.py index c48a733..afbe047 100644 --- a/intersimple-expert-rollout-setobs2.py +++ b/intersimple-expert-rollout-setobs2.py @@ -56,6 +56,10 @@ def main(track:int, loc:int=0): torch.save(expert_data, f'intersimple-expert-data-setobs2-loc{loc}-track{track}.pt') +def loop(tracks:list=[0]): + for track in tracks: + main(track) + if __name__=='__main__': import fire - fire.Fire(main) \ No newline at end of file + fire.Fire(loop) \ No newline at end of file