1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
from yacs.config import CfgNode as CN
from torch_geometric.graphgym.register import register_config
@register_config('example')
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'
|
