microsoft
/
MMdnn
зеркало из https://github.com/microsoft/MMdnn.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

nit

This commit is contained in:
Kit 2018-02-09 13:47:34 +08:00
Родитель c4aba456d0
Коммит 345265be54
1 изменённых файлов: 10 добавлений и 10 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

20
tests/test_conversion_imagenet.py
Просмотреть файл

@ -305,19 +305,19 @@ class TestModels(CorrectnessTest):
'nasnet' : [TensorflowEmit, KerasEmit],
},
'mxnet' : {
# 'vgg19' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
# 'imagenet1k-inception-bn' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
# 'imagenet1k-resnet-152' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'vgg19' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'imagenet1k-inception-bn' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'imagenet1k-resnet-152' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'squeezenet_v1.1' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
# 'imagenet1k-resnext-101-64x4d' : [CntkEmit, TensorflowEmit, PytorchEmit], # Keras is too slow
# 'imagenet1k-resnext-50' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
'imagenet1k-resnext-101-64x4d' : [CntkEmit, TensorflowEmit, PytorchEmit], # Keras is too slow
'imagenet1k-resnext-50' : [CntkEmit, TensorflowEmit, KerasEmit, PytorchEmit],
},
'caffe' : {
'vgg19' : [KerasEmit],
# 'alexnet' : [KerasEmit],
'inception_v1' : [CntkEmit],
'resnet152' : [CntkEmit],
'squeezenet' : [CntkEmit]
# 'vgg19' : [KerasEmit],
'alexnet' : [TensorflowEmit, KerasEmit],
# 'inception_v1' : [CntkEmit],
# 'resnet152' : [CntkEmit],
# 'squeezenet' : [CntkEmit]
},
'tensorflow' : {
'inception_v1' : [TensorflowEmit, KerasEmit, PytorchEmit], # TODO: CntkEmit