Skip to content
Snippets Groups Projects
Commit 3ebb418e authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

switch args order for dropout

parent d8750df4
No related branches found
No related tags found
No related merge requests found
Pipeline #82126 failed
...@@ -29,7 +29,7 @@ using namespace Aidge; ...@@ -29,7 +29,7 @@ using namespace Aidge;
}); });
constexpr const float dropout_prob = 0.5f; constexpr const float dropout_prob = 0.5f;
std::shared_ptr<Node> myDropout = Dropout(dropout_prob, Dropout_Op::DEFAULT_SEED, true); // assumes it is training mode std::shared_ptr<Node> myDropout = Dropout(dropout_prob, true, Dropout_Op::DEFAULT_SEED); // assumes it is training mode
auto op = std::static_pointer_cast<OperatorTensor>(myDropout->getOperator()); auto op = std::static_pointer_cast<OperatorTensor>(myDropout->getOperator());
op->associateInput(0, input); op->associateInput(0, input);
...@@ -108,7 +108,7 @@ using namespace Aidge; ...@@ -108,7 +108,7 @@ using namespace Aidge;
}); });
constexpr float dropout_prob = 0.5f; constexpr float dropout_prob = 0.5f;
std::shared_ptr<Node> myDropout = Dropout(dropout_prob, Dropout_Op::DEFAULT_SEED,true); std::shared_ptr<Node> myDropout = Dropout(dropout_prob, true, Dropout_Op::DEFAULT_SEED);
auto op = std::static_pointer_cast<OperatorTensor>(myDropout->getOperator()); auto op = std::static_pointer_cast<OperatorTensor>(myDropout->getOperator());
op->associateInput(0, input); op->associateInput(0, input);
op->setBackend("cpu"); op->setBackend("cpu");
...@@ -162,7 +162,7 @@ using namespace Aidge; ...@@ -162,7 +162,7 @@ using namespace Aidge;
}); });
// Run 1 // Run 1
std::shared_ptr<Node> dropout1 = Dropout(dropout_prob, fixed_seed, true); std::shared_ptr<Node> dropout1 = Dropout(dropout_prob, true, fixed_seed);
auto op1 = std::static_pointer_cast<OperatorTensor>(dropout1->getOperator()); auto op1 = std::static_pointer_cast<OperatorTensor>(dropout1->getOperator());
op1->associateInput(0, input); op1->associateInput(0, input);
op1->setBackend("cpu"); op1->setBackend("cpu");
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment