зеркало из https://github.com/microsoft/archai.git
additional test for config resolution and override
This commit is contained in:
Родитель
6be9f214ab
Коммит
1d42b62d4f
|
@ -1,6 +1,6 @@
|
||||||
from archai.common.config import Config
|
from archai.common.config import Config
|
||||||
|
|
||||||
def test_param_override():
|
def test_param_override1():
|
||||||
conf = Config('confs/algos/darts.yaml;confs/datasets/cifar10.yaml')
|
conf = Config('confs/algos/darts.yaml;confs/datasets/cifar10.yaml')
|
||||||
assert not conf['nas']['eval']['trainer']['apex']['enabled']
|
assert not conf['nas']['eval']['trainer']['apex']['enabled']
|
||||||
assert not conf['nas']['eval']['loader']['apex']['enabled']
|
assert not conf['nas']['eval']['loader']['apex']['enabled']
|
||||||
|
@ -10,4 +10,15 @@ def test_param_override():
|
||||||
assert conf['nas']['eval']['trainer']['apex']['enabled']
|
assert conf['nas']['eval']['trainer']['apex']['enabled']
|
||||||
assert conf['nas']['eval']['loader']['apex']['enabled']
|
assert conf['nas']['eval']['loader']['apex']['enabled']
|
||||||
|
|
||||||
test_param_override()
|
def test_param_override2():
|
||||||
|
conf = Config('confs/algos/darts.yaml;confs/datasets/imagenet.yaml')
|
||||||
|
assert conf['nas']['eval']['trainer']['apex']['distributed_enabled']
|
||||||
|
assert conf['nas']['eval']['loader']['apex']['distributed_enabled']
|
||||||
|
|
||||||
|
conf = Config('confs/algos/darts.yaml;confs/datasets/imagenet.yaml',
|
||||||
|
param_args=["--nas.eval.trainer.apex.enabled", "True", "--nas.eval.trainer.apex.distributed_enabled", "False"])
|
||||||
|
assert not conf['nas']['eval']['trainer']['apex']['distributed_enabled']
|
||||||
|
assert not conf['nas']['eval']['loader']['apex']['distributed_enabled']
|
||||||
|
|
||||||
|
test_param_override1()
|
||||||
|
test_param_override2()
|
Загрузка…
Ссылка в новой задаче