parser.add_argument("--oneshot_input", type=str, default="head")
-parser.add_argument("--oneshot_output", type=str, default="policy")
+parser.add_argument("--oneshot_output", type=str, default="trace")
######################################################################