Generate dataset for video prediction

2026-09-11 11:53:01 +08:00 · 2017-12-16 20:03:11 -07:00
parent 56304e63e6
commit 214f7693c2
8 changed files with 154 additions and 18 deletions
@@ -267,6 +267,9 @@ def d3pg_continuous():
    agent.run()

 if __name__ == '__main__':
+    mkdir('data')
+    mkdir('log')
+    os.system('export OMP_NUM_THREADS=1')
    # logger.setLevel(logging.DEBUG)
    logger.setLevel(logging.INFO)