Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
monticore
EmbeddedMontiArc
generators
EMADL2CPP
Commits
fc007f99
Commit
fc007f99
authored
Jul 20, 2019
by
Nicola Gatto
Browse files
Add trained architecture checks
parent
e9c67a96
Changes
1
Hide whitespace changes
Inline
Side-by-side
src/main/java/de/monticore/lang/monticar/emadl/generator/EMADLGenerator.java
View file @
fc007f99
...
...
@@ -34,6 +34,8 @@ import de.monticore.lang.monticar.cnnarch._symboltable.ArchitectureSymbol;
import
de.monticore.lang.monticar.cnnarch._symboltable.SerialCompositeElementSymbol
;
import
de.monticore.lang.monticar.cnnarch.gluongenerator.CNNTrain2Gluon
;
import
de.monticore.lang.monticar.cnnarch.gluongenerator.annotations.ArchitectureAdapter
;
import
de.monticore.lang.monticar.cnntrain._cocos.CNNTrainCoCoChecker
;
import
de.monticore.lang.monticar.cnntrain._cocos.CNNTrainCocos
;
import
de.monticore.lang.monticar.cnntrain._symboltable.ConfigurationSymbol
;
import
de.monticore.lang.monticar.emadl._cocos.EMADLCocos
;
import
de.monticore.lang.monticar.generator.FileContent
;
...
...
@@ -536,6 +538,7 @@ public class EMADLGenerator {
assert
correspondingArchitecture
!=
null
:
"No architecture found for train "
+
fullConfigName
+
" configuration!"
;
configuration
.
setTrainedArchitecture
(
new
ArchitectureAdapter
(
correspondingArchitecture
.
getName
(),
correspondingArchitecture
));
CNNTrainCocos
.
checkTrainedArchitectureCoCos
(
configuration
);
// Resolve critic network if critic is present
if
(
configuration
.
getCriticName
().
isPresent
())
{
...
...
@@ -556,6 +559,7 @@ public class EMADLGenerator {
}
critic
.
get
().
setComponentName
(
fullCriticName
);
configuration
.
setCriticNetwork
(
new
ArchitectureAdapter
(
fullCriticName
,
critic
.
get
()));
CNNTrainCocos
.
checkCriticCocos
(
configuration
);
}
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment