1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
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)
|
