예제 #1
0
 def test_input_names(onnx_model, path):
     initializer_names = get_initializer_names(onnx_model)
     assert len(initializer_names) == 4
     assert 'FixedBatchNormalization_0_param_avg_mean' in\
         initializer_names
     assert 'FixedBatchNormalization_0_param_avg_var' in\
         initializer_names
예제 #2
0
 def test_input_names(onnx_model, path):
     initializer_names = get_initializer_names(onnx_model)
     assert len(initializer_names) == 4
     assert 'param_bn_avg_mean' in initializer_names
     assert 'param_bn_avg_var' in initializer_names