Added replay for LargeMemory layer

parent 047ac311
Pipeline #284484 failed with stage
in 1 minute and 27 seconds
......@@ -221,24 +221,40 @@ public class ArchitectureElementData {
return getLayerSymbol().getIntValue(AllPredefinedLayers.REPLAY_STEPS_NAME).get();
}
public int getReplayGradientStepsTraining(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.REPLAY_GRADIENT_STEPS_TRAINING_NAME).get();
public int getReplayGradientSteps(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.REPLAY_GRADIENT_STEPS_NAME).get();
}
public double getStoreProb(){
return getLayerSymbol().getDoubleValue(AllPredefinedLayers.STORE_PROB_NAME).get();
public double getReplayMemoryStoreProb(){
return getLayerSymbol().getDoubleValue(AllPredefinedLayers.REPLAY_MEMORY_STORE_PROB_NAME).get();
}
public int getMaxStoredSamples(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.MAX_STORED_SAMPLES_NAME).get();
}
public int getReplayK(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.REPLAY_K_NAME).get();
public boolean getUseReplay(){
return getLayerSymbol().getBooleanValue(AllPredefinedLayers.USE_REPLAY_NAME).get();
}
public int getReplayGradientStepsPrediction(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.REPLAY_GRADIENT_STEPS_PREDICTION_NAME).get();
public boolean getUseLocalAdaption(){
return getLayerSymbol().getBooleanValue(AllPredefinedLayers.USE_LOCAL_ADAPTION_NAME).get();
}
public String getReplayMemoryStoreDistMeasure(){
return getLayerSymbol().getStringValue(AllPredefinedLayers.REPLAY_MEMORY_STORE_DIST_MEASURE_NAME).get();
}
public String getLocalAdaptionMemoryStoreDistMeasure(){
return getLayerSymbol().getStringValue(AllPredefinedLayers.LOCAL_ADAPTION_MEMORY_STORE_DIST_MEASURE_NAME).get();
}
public int getLocalAdaptionK(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.LOCAL_ADAPTION_K_NAME).get();
}
public int getlocalAdaptionGradientSteps(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.LOCAL_ADAPTION_GRADIENT_STEPS_NAME).get();
}
public String getQueryNetDir(){
......
......@@ -5,6 +5,8 @@ import de.monticore.lang.monticar.cnnarch._symboltable.*;
import de.monticore.lang.monticar.cnnarch.predefined.Convolution;
import de.monticore.lang.monticar.cnnarch.predefined.FullyConnected;
import de.monticore.lang.monticar.cnnarch.predefined.Pooling;
import de.monticore.lang.monticar.cnnarch.predefined.Memory;
import de.monticore.lang.monticar.cnnarch.predefined.ReplayMemory;
import java.util.*;
......@@ -132,6 +134,8 @@ public class LayerNameCreator {
return "fc";
} else if (layerDeclaration instanceof Pooling) {
return "pool";
} else if (layerDeclaration instanceof Memory || layerDeclaration instanceof ReplayMemory) {
return "memory";
} else {
return layerDeclaration.getName().toLowerCase();
}
......
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