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

@@ -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)