removing demo generation script and including it in readme. updating expert data generator to loop over tracks and match readme

This commit is contained in:
Arec Jamgochian
2022-03-03 20:20:36 -08:00
parent 37f44605d2
commit 3d3b3d510a
3 changed files with 5 additions and 11 deletions

View File

@@ -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]'

Binary file not shown.

View File

@@ -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)
fire.Fire(loop)