Exemplos de match_parameters em Python

Exemplo n.º 1

Arquivo: optim_helper_test.py Projeto: celidos/TensorRT_study

 def test_multi_keys(self, resnet18):
     param = list(helper.match_parameters(resnet18,
                                          ['conv1', 'downsample']))
     assert len(param) == 18

Exemplo n.º 2

Arquivo: optim_helper_test.py Projeto: celidos/TensorRT_study

    def test_regex(self, resnet18):
        param = helper.match_parameters(resnet18, ['downsample.*.weight$'])
        assert len(list(param)) == 6

        param = helper.match_parameters(resnet18, ['downsample.*.wei$'])
        assert not list(param)

Exemplo n.º 3

Arquivo: optim_helper_test.py Projeto: celidos/TensorRT_study

 def test_single_key(self, resnet18):
     param = helper.match_parameters(resnet18, ['downsample.0.weight'])
     assert len(list(param)) == 3