Commit 06e9f2b6 authored by Sebastian Nickels's avatar Sebastian Nickels

Changed TrainParamSupportChecker

parent aff7ca07
......@@ -9,9 +9,7 @@ import de.se_rwth.commons.logging.Log;
import java.util.List;
public class ArchitectureSupportChecker {
public ArchitectureSupportChecker() {}
public abstract class ArchitectureSupportChecker {
// Overload functions returning always true to enable the features
protected boolean checkMultipleStreams(ArchitectureSymbol architecture) {
......
......@@ -6,19 +6,19 @@ import de.se_rwth.commons.logging.Log;
import java.util.ArrayList;
import java.util.List;
public class TrainParamSupportChecker implements CNNTrainVisitor {
public abstract class TrainParamSupportChecker implements CNNTrainVisitor {
private List<String> unsupportedElemList = new ArrayList<>();
protected List<String> unsupportedElemList = new ArrayList<>();
private void printUnsupportedEntryParam(String nodeName){
protected void printUnsupportedEntryParam(String nodeName){
Log.warn("Unsupported training parameter " + "'" + nodeName + "'" + " for the current backend. It will be ignored.");
}
private void printUnsupportedOptimizer(String nodeName){
protected void printUnsupportedOptimizer(String nodeName){
Log.warn("Unsupported optimizer parameter " + "'" + nodeName + "'" + " for the current backend. It will be ignored.");
}
private void printUnsupportedOptimizerParam(String nodeName){
protected void printUnsupportedOptimizerParam(String nodeName){
Log.warn("Unsupported training optimizer parameter " + "'" + nodeName + "'" + " for the current backend. It will be ignored.");
}
......
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