Commit 8798cd10 authored by Julian Treiber's avatar Julian Treiber

added load_pretrained parameter

parent 7f595808
......@@ -64,6 +64,13 @@ public class ConfigurationData {
return String.valueOf(getConfiguration().getEntry("log_period").getValue());
}
public Boolean getLoadPretrained() {
if (!getConfiguration().getEntryMap().containsKey("load_pretrained")) {
return null;
}
return (Boolean) getConfiguration().getEntry("load_pretrained").getValue().getValue();
}
public Boolean getNormalize() {
if (!getConfiguration().getEntryMap().containsKey("normalize")) {
return null;
......@@ -128,7 +135,7 @@ public class ConfigurationData {
} else{
return mapToStrings;}
}
public String getLossWeights() {
if (!getConfiguration().getEntryMap().containsKey("loss_weights")) {
return null;
......
......@@ -40,12 +40,14 @@ public abstract class TrainParamSupportChecker implements CNNTrainVisitor {
public void visit(ASTLoadCheckpointEntry node){}
public void visit(ASTLoadPretrainedEntry node){}
public void visit(ASTNormalizeEntry node){}
public void visit(ASTTrainContextEntry node){}
public void visit(ASTEvalMetricEntry node){}
public void visit(ASTSGDOptimizer node){}
public void visit(ASTAdamOptimizer node){}
......
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