-
Notifications
You must be signed in to change notification settings - Fork 23
/
example.py
26 lines (17 loc) · 712 Bytes
/
example.py
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
26
from yacs.config import CfgNode as CN
from graphgym.register import register_config
def set_cfg_example(cfg):
r'''
This function sets the default config value for customized options
:return: customized configuration use by the experiment.
'''
# ------------------------------------------------------------------------ #
# Customized options
# ------------------------------------------------------------------------ #
# example argument
cfg.example_arg = 'example'
# example argument group
cfg.example_group = CN()
# then argument can be specified within the group
cfg.example_group.example_arg = 'example'
register_config('example', set_cfg_example)