diff --git a/mmdnn/conversion/cntk/cntk_emitter.py b/mmdnn/conversion/cntk/cntk_emitter.py index 6778d399..770164c7 100644 --- a/mmdnn/conversion/cntk/cntk_emitter.py +++ b/mmdnn/conversion/cntk/cntk_emitter.py @@ -766,7 +766,7 @@ def yolo_parameter(): '''.format(self.yolo_parameter)) - def _layer_upsample(self): + def _layer_UpSampling2D(self): self.add_body(0, ''' def Upsampling2D(x, stride, name): assert stride == 2