Skip to content
Snippets Groups Projects

Scheduler backprop

Merged Maxence Naud requested to merge scheduler_backprop into dev
1 file
+ 4
3
Compare changes
  • Side-by-side
  • Inline
@@ -20,7 +20,7 @@
#include "aidge/scheduler/Scheduler.hpp"
#include "aidge/backend/cpu.hpp"
#include "aidge/recipies/GraphViewHelper.hpp"
#include "aidge/recipes/GraphViewHelper.hpp"
using namespace Aidge;
@@ -372,7 +372,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
compile_gradient(gv);
SequentialScheduler scheduler(gv);
scheduler.forward();
auto predictedOutput = gv->getOrderedOutputs()[0];
auto predictedOutput = gv->getOrderedOutputs()[0].first;
std::shared_ptr<Tensor> targetOutput =
std::make_shared<Tensor>(Array4D<float, 2, 1, 5, 5>{{{{{0.0f, 1.0f, 1.0f, 2.0f, 2.0f},
@@ -386,4 +386,5 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
{6.0f, 6.0f, 6.0f, 7.0f, 7.0f},
{7.0f, 7.0f, 7.0f, 7.0f, 7.0f}}}}});
REQUIRE_NOTHROW(scheduler.backward(predictedOutput - targetOutput));
REQUIRE_NOTHROW(scheduler.backward({targetOutput}));
}
\ No newline at end of file
Loading