We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 28a810a commit fdbf106Copy full SHA for fdbf106
apps/and.cpp
@@ -11,7 +11,7 @@ int main(int argc, char** argv) {
11
if (argc > 1 && std::string(argv[1]) == "debug") Logger::get_instance().set_debug_mode(true);
12
else if (argc > 1) throw std::runtime_error("Invalid argument in main function. Use 'debug' to enable debug mode.");
13
14
- int num_epochs = 500000, num_samples = 4;
+ int num_epochs = 50000, num_samples = 4;
15
float learning_rate = 0.001;
16
17
// weights shared pointer
@@ -51,7 +51,7 @@ int main(int argc, char** argv) {
51
loss = cross_entropy(Y[j], a) + loss;
52
}
53
54
- // loss = loss / Scalar<double>::make(num_samples);
+ loss = loss / Scalar<double>::make(num_samples);
55
loss->backward();
56
57
// update weights
0 commit comments