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:
@@ -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.
@@ -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')
|
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__':
|
if __name__=='__main__':
|
||||||
import fire
|
import fire
|
||||||
fire.Fire(main)
|
fire.Fire(loop)
|
||||||
Reference in New Issue
Block a user