Added BroadcastAdd and Multiply layer. Renamed Multiply layer to...
Added BroadcastAdd and Multiply layer. Renamed Multiply layer to BroadcastMultiply. Added axis parameter to Softmax layer. Added SoftmaxCrossEntropyLossIgnoreIndices to SupervisedTrainer for now, in order to ignore certain words/indices for loss computation in NLP applications
parent
95a67ac4
No related branches found
No related tags found
Showing
- src/main/java/de/monticore/lang/monticar/cnnarch/gluongenerator/CNNArch2GluonLayerSupportChecker.java 8 additions, 0 deletions...arch/gluongenerator/CNNArch2GluonLayerSupportChecker.java
- src/main/resources/templates/gluon/CNNNet.ftl 28 additions, 2 deletionssrc/main/resources/templates/gluon/CNNNet.ftl
- src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl 59 additions, 2 deletionssrc/main/resources/templates/gluon/CNNSupervisedTrainer.ftl
- src/main/resources/templates/gluon/elements/BroadcastAdd.ftl 8 additions, 0 deletionssrc/main/resources/templates/gluon/elements/BroadcastAdd.ftl
- src/main/resources/templates/gluon/elements/BroadcastMultiply.ftl 6 additions, 0 deletions.../resources/templates/gluon/elements/BroadcastMultiply.ftl
- src/main/resources/templates/gluon/elements/Multiply.ftl 0 additions, 10 deletionssrc/main/resources/templates/gluon/elements/Multiply.ftl
- src/main/resources/templates/gluon/elements/Reshape.ftl 9 additions, 0 deletionssrc/main/resources/templates/gluon/elements/Reshape.ftl
- src/main/resources/templates/gluon/elements/Softmax.ftl 2 additions, 1 deletionsrc/main/resources/templates/gluon/elements/Softmax.ftl
Loading
Please register or sign in to comment