New features
This commit is contained in:
parent
9397934825
commit
d52bc702a2
|
@ -70,6 +70,9 @@ def set_cfg(explaining_cfg):
|
|||
# Set wether or not load the best model for given dataset or a path
|
||||
explaining_cfg.model.ckpt = "best"
|
||||
|
||||
# Setting the path of models folder
|
||||
explaining_cfg.model.path = "path"
|
||||
|
||||
# ----------------------------------------------------------------------- #
|
||||
# Explainer options
|
||||
# ----------------------------------------------------------------------- #
|
||||
|
|
|
@ -3,7 +3,7 @@ import argparse
|
|||
|
||||
def parse_args() -> argparse.Namespace:
|
||||
r"""Parses the command line arguments."""
|
||||
parser = argparse.ArgumentParser(description="GraphGym")
|
||||
parser = argparse.ArgumentParser(description="ExplainingFramework")
|
||||
|
||||
parser.add_argument(
|
||||
"--cfg",
|
||||
|
|
|
@ -1,5 +1,14 @@
|
|||
class Explaining(object):
|
||||
def __init__(self, cfg: dict, explaining_cfg: dict, explainer_cfg: dict = None):
|
||||
self.cfg = cfg
|
||||
class ExplainingOutline(object):
|
||||
def __init__(self, explaining_cfg: CN, explainer_cfg: CN = None):
|
||||
self.explaining_cfg = explaining_cfg
|
||||
self.explainer_cfg = explainer_cfg
|
||||
|
||||
|
||||
|
||||
def load_cfg(self):
|
||||
if self
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue