added functionality for ArgMax layer
parent
f337232c
No related branches found
No related tags found
Showing
- src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl 23 additions, 12 deletionssrc/main/resources/templates/gluon/CNNSupervisedTrainer.ftl
- src/main/resources/templates/gluon/elements/ArgMax.ftl 2 additions, 1 deletionsrc/main/resources/templates/gluon/elements/ArgMax.ftl
- src/main/resources/templates/gluon/pythonExecute.ftl 10 additions, 0 deletionssrc/main/resources/templates/gluon/pythonExecute.ftl
- src/test/resources/target_code/CNNSupervisedTrainer_Alexnet.py 18 additions, 8 deletions...est/resources/target_code/CNNSupervisedTrainer_Alexnet.py
- src/test/resources/target_code/CNNSupervisedTrainer_CifarClassifierNetwork.py 18 additions, 8 deletions...arget_code/CNNSupervisedTrainer_CifarClassifierNetwork.py
- src/test/resources/target_code/CNNSupervisedTrainer_VGG16.py 18 additions, 8 deletionssrc/test/resources/target_code/CNNSupervisedTrainer_VGG16.py
Loading
Please register or sign in to comment