1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import argparse
from torch_geometric.graphgym.utils.agg_runs import agg_batch
def parse_args():
"""Parses the arguments."""
parser = argparse.ArgumentParser(
description='Train a classification model')
parser.add_argument('--dir', dest='dir', help='Dir for batch of results',
required=True, type=str)
parser.add_argument('--metric', dest='metric',
help='metric to select best epoch', required=False,
type=str, default='auto')
return parser.parse_args()
args = parse_args()
agg_batch(args.dir, args.metric)
|
