Fixing bugs and adding new features

This commit is contained in:
araison 2023-01-02 23:37:40 +01:00
parent db0be0ddb7
commit 9ad5adb33e
13 changed files with 183 additions and 120 deletions

View file

@ -114,7 +114,7 @@ def set_cfg(explaining_cfg):
explaining_cfg.threshold_config.threshold_type = None
explaining_cfg.threshold_config.value = [0.3, 0.5, 0.7]
explaining_cfg.threshold_config.value = [i * 0.05 for i in range(21)]
explaining_cfg.threshold_config.relu_and_normalize = True
@ -128,8 +128,7 @@ def set_cfg(explaining_cfg):
explaining_cfg.metrics.force = False
explaining_cfg.attack = CN()
explaining_cfg.attack.name = 'all'
explaining_cfg.attack.name = "all"
explaining_cfg.accelerator = "auto"