1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
import argparse
import torch
def dump(filename):
schemas = torch._C._jit_get_all_schemas()
schemas += torch._C._jit_get_custom_class_schemas()
with open(filename, "w") as f:
for s in schemas:
f.write(str(s))
f.write("\n")
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Process some integers.")
parser.add_argument(
"-f",
"--filename",
help="filename to dump the schemas",
type=str,
default="schemas.txt",
)
args = parser.parse_args()
dump(args.filename)
|
