This commit is contained in:
Ru ZHANG 2018-02-09 13:54:03 +08:00
Родитель 9833b75050
Коммит 58c9e22bf0
1 изменённых файлов: 5 добавлений и 5 удалений

Просмотреть файл

@ -301,11 +301,11 @@ class TestModels(CorrectnessTest):
'imagenet1k-resnext-50' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
},
'caffe' : {
'vgg19' : [KerasEmit],
# 'alexnet' : [KerasEmit],
'inception_v1' : [CntkEmit],
'resnet152' : [CntkEmit],
'squeezenet' : [CntkEmit]
'vgg19' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'alexnet' : [CntkEmit],
'inception_v1' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'resnet152' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'squeezenet' : [CntkEmit, PytorchEmit]
},
'tensorflow' : {
'inception_v1' : [TensorflowEmit, KerasEmit, PytorchEmit], # TODO: CntkEmit