Commit 28de2022 authored by Julian Dierkes's avatar Julian Dierkes

extended for GANs

parent 120f39f4
......@@ -227,7 +227,7 @@ public class ArchitectureElementData {
return getTransPadding(getLayerSymbol()); //The padding calculated here is only used in the gluon/ mxnet backend, in the tensorlflow one it is interpreted as "same"
}else if(pad.equals("valid")){
return Arrays.asList(0,0);
}else{ //"no loss"
}else{ //"no loss" //todo remove
return Arrays.asList(0,0,-1,0,0,0,0,0);
}
}
......
......@@ -50,6 +50,34 @@ public class ConfigurationData {
return String.valueOf(getConfiguration().getEntry("k_value") .getValue());
}
public String getGeneratorLossWeight() {
if (!getConfiguration().getEntryMap().containsKey("generator_loss_weight")) {
return null;
}
return String.valueOf(getConfiguration().getEntry("generator_loss_weight") .getValue());
}
public String getDiscriminatorLossWeight() {
if (!getConfiguration().getEntryMap().containsKey("discriminator_loss_weight")) {
return null;
}
return String.valueOf(getConfiguration().getEntry("discriminator_loss_weight") .getValue());
}
public String getSpeedPeriod() {
if (!getConfiguration().getEntryMap().containsKey("speed_period")) {
return null;
}
return String.valueOf(getConfiguration().getEntry("speed_period") .getValue());
}
public Boolean getPrintImages() {
if (!getConfiguration().getEntryMap().containsKey("print_images")) {
return null;
}
return (Boolean) getConfiguration().getEntry("print_images").getValue().getValue();
}
public String getGeneratorLoss() {
if (!getConfiguration().getEntryMap().containsKey("generator_loss")) {
return null;
......@@ -57,11 +85,11 @@ public class ConfigurationData {
return String.valueOf(getConfiguration().getEntry("generator_loss") .getValue());
}
public String getConditionalInput() {
if (!getConfiguration().getEntryMap().containsKey("conditional_input")) {
public String getGeneratorTargetName() {
if (!getConfiguration().getEntryMap().containsKey("generator_target_name")) {
return null;
}
return String.valueOf(getConfiguration().getEntry("conditional_input") .getValue());
return String.valueOf(getConfiguration().getEntry("generator_target_name") .getValue());
}
public String getNoiseInput() {
......@@ -120,6 +148,10 @@ public class ConfigurationData {
return (String) getConfiguration().getEntry("preprocessing_name").getValue().toString();
}
public Boolean getPreprocessor() {
return (Boolean) configuration.hasPreprocessor();
}
public String getContext() {
if (!getConfiguration().getEntryMap().containsKey("context")) {
return null;
......
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