araison/explaining_framework
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Adding new features

Browse source
This commit is contained in:
araison 2022-12-29 23:29:32 +01:00
parent e2d47af072
commit a00e73d4f0
6 changed files with 103 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

9
explaining_framework/config/explaining_config.py
View file

@ -119,15 +119,20 @@ def set_cfg(explaining_cfg):
explaining_cfg.threshold_config.relu_and_normalize = True
# Select device: 'cpu', 'cuda', 'auto'
explaining_cfg.accelerator = "auto"
# which objectives metrics to computes, either all or one in particular if implemented
explaining_cfg.metrics = CN()
explaining_cfg.metrics.type = "all"
explaining_cfg.metrics.name = "all"
# Whether or not recomputing metrics if they already exist
explaining_cfg.metrics.force = False
explaining_cfg.attack = CN()
explaining_cfg.attack.name = 'all'
explaining_cfg.accelerator = "auto"
def assert_cfg(explaining_cfg):
r"""Checks config values, do necessary post processing to the configs"""