-
Notifications
You must be signed in to change notification settings - Fork 189
Closed
Description
We are going to support the conversion of models as follows at first stage. They all can be found in our models bank, some maybe only need to be verified after #19 merged, and some lack of necessary operators.
- recognize_digits ( Support conversion of recognize_digits models #19 @kuke)
- ResNet50 (Enhance batch_norm_op & rename output args sharing names with inputs #30 Add vgg16 & resnet50 to supported models #32 @kuke)
- VGG16 (Enhance batch_norm_op & rename output args sharing names with inputs #30 Add vgg16 & resnet50 to supported models #32 @kuke)
- MobileNet (Enhance batch_norm_op & rename output args sharing names with inputs #30 Add mobilenet & se_resnext to supported models #33 @kuke)
- SE-ResNeXt (Enhance batch_norm_op & rename output args sharing names with inputs #30 Add mobilenet & se_resnext to supported models #33 @kuke)
- Inception-v4 (first add the Fluid config in models/fluid/image_classification) (#900 #49 @kuke)
Even all the models above are supported, only a small subset of operators are used and verified. So for the rest operators, we may need another task:
- Design the unit test framework to test every operator's conversion (Add unit test framework for operators' conversion #29 @kuke)
Hints:
- Choose one task every time and assign yourself when you are really ready for working on it;
- Please append the model to the supported models section in README like Support conversion of recognize_digits models #19 after its conversion being validated.
pkuyym, sidgoyal78 and luotao1
Metadata
Metadata
Assignees
Labels
No labels