Remove fixing the dimension manually

Signed-off-by: Omar Shrit <omar@avontech.fr>
2024-03-21 07:11:58 +01:00
parent 01d0bc840b
commit 2f71e925d0
1 changed files with 1 additions and 1 deletions
@@ -117,7 +117,7 @@ TEST_CASE("PReLUIntegrationTest", "[ANNLayerTest]")
  model.Add<Linear>(3);
  model.Add<PReLU>(0.01);
  model.Add<Linear>(1);
-  model.InputDimensions() = std::vector<size_t>({trainData.n_rows});
+
  // Sometimes the model may not optimize correctly, so we allow a few trials.
  bool success = false;
  for (size_t trial = 0; trial < 3; ++trial)