Commit ec3f56ed authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Fixed tests and reverted _label change since it is necessary for MXNet

parent 1812de35
Pipeline #144203 failed with stages
in 3 minutes and 16 seconds
......@@ -7,7 +7,7 @@ from mxnet import nd
class ${tc.fileNameWithoutEnding}:
_input_names_ = [${tc.join(tc.architectureInputs, ",", "'", "'")}]
_output_names_ = [${tc.join(tc.architectureOutputs, ",", "'", "'")}]
_output_names_ = [${tc.join(tc.architectureOutputs, ",", "'", "_label'")}]
def __init__(self):
self._data_dir = "${tc.dataPath}/"
......
......@@ -127,7 +127,7 @@ public class GenerationTest extends AbstractSymtabTest {
Log.getFindings().clear();
String[] args = {"-m", "src/test/resources/invalid_tests", "-r", "MultipleOutputs"};
CNNArch2GluonCli.main(args);
assertTrue(Log.getFindings().size() == 2);
assertTrue(Log.getFindings().size() == 3);
}
@Test
......@@ -135,7 +135,7 @@ public class GenerationTest extends AbstractSymtabTest {
Log.getFindings().clear();
String[] args = {"-m", "src/test/resources/invalid_tests", "-r", "MultipleStreams"};
CNNArch2GluonCli.main(args);
assertTrue(Log.getFindings().size() == 2);
assertTrue(Log.getFindings().size() == 1);
}
@Test
......
......@@ -7,7 +7,7 @@ from mxnet import nd
class CNNDataLoader_Alexnet:
_input_names_ = ['data']
_output_names_ = ['predictions']
_output_names_ = ['predictions_label']
def __init__(self):
self._data_dir = "data/Alexnet/"
......
......@@ -7,7 +7,7 @@ from mxnet import nd
class CNNDataLoader_CifarClassifierNetwork:
_input_names_ = ['data']
_output_names_ = ['softmax']
_output_names_ = ['softmax_label']
def __init__(self):
self._data_dir = "data/CifarClassifierNetwork/"
......
......@@ -7,7 +7,7 @@ from mxnet import nd
class CNNDataLoader_VGG16:
_input_names_ = ['data']
_output_names_ = ['predictions']
_output_names_ = ['predictions_label']
def __init__(self):
self._data_dir = "data/VGG16/"
......
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