search

undertherain / benchmarker

modular framework for [not only] deep learning performance benchmarking
http://blackbird.pw/performance
Mozilla Public License 2.0
9 stars 5 forks source link

mxnet framework returns error with resnet50 & vgg16 #172

Closed shwetasalaria closed 3 years ago

shwetasalaria commented 3 years ago

Case 1: python3 -m benchmarker --framework=mxnet --problem=resnet50 --problem_size=512 --batch_size=32 --mode=training --nb_epoch=1

Error: KeyError: 'channels_first' in "/benchmarker/data/synthetic/helpers.py", line 10, in set_image_size

Case 2: python3 -m benchmarker --framework=mxnet --problem=vgg16 --problem_size=512 --batch_size=32 --mode=training --nb_epoch=1

Error: AttributeError: module 'benchmarker.kernels.vgg16.mxnet' has no attribute 'get_kernel'