Commit 2120b7f7 authored by Cihad's avatar Cihad
Browse files

Refactoring - final 2

parent a629a872
Pipeline #519320 passed with stages
in 3 minutes and 33 seconds
......@@ -6,7 +6,10 @@ import com.google.common.collect.Maps;
import conflang._symboltable.ConfLangConfigurationSymbol;
import de.monticore.lang.monticar.cnnarch.generator.CNNTrainGenerator;
import de.monticore.lang.monticar.cnnarch.generator.TemplateConfiguration;
import de.monticore.lang.monticar.cnnarch.generator.training.CNNTrainingConfiguration;
import de.monticore.lang.monticar.cnnarch.generator.training.TrainingComponentsContainer;
import de.monticore.lang.monticar.cnnarch.generator.validation.gan.GANChecker;
import de.monticore.lang.monticar.cnnarch.generator.validation.reinforcement.ReinforcementLearningChecker;
import de.monticore.lang.monticar.generator.FileContent;
import schemalang._symboltable.SchemaLangDefinitionSymbol;
......@@ -29,6 +32,10 @@ public class CNNTrain2MxNet extends CNNTrainGenerator {
validateConfiguration(rootModelName, configuration, schema);
CNNTrainingConfiguration.initialize(configuration); // TODO move these calls to super class
GANChecker.check();
ReinforcementLearningChecker.check();
generateFilesFromConfigurationSymbol(configuration, schema, new TrainingComponentsContainer());
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment