parser.add_argument("--oneshot", action="store_true", default=False)
parser.add_argument("--oneshot_input", type=str, default="head")
parser.add_argument("--oneshot", action="store_true", default=False)
parser.add_argument("--oneshot_input", type=str, default="head")