parser.add_argument("--device", type=str, default="cuda:0")