Skip to content

Commit fdbf106

Browse files
thread sanitizer
1 parent 28a810a commit fdbf106

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

apps/and.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ int main(int argc, char** argv) {
1111
if (argc > 1 && std::string(argv[1]) == "debug") Logger::get_instance().set_debug_mode(true);
1212
else if (argc > 1) throw std::runtime_error("Invalid argument in main function. Use 'debug' to enable debug mode.");
1313

14-
int num_epochs = 500000, num_samples = 4;
14+
int num_epochs = 50000, num_samples = 4;
1515
float learning_rate = 0.001;
1616

1717
// weights shared pointer
@@ -51,7 +51,7 @@ int main(int argc, char** argv) {
5151
loss = cross_entropy(Y[j], a) + loss;
5252
}
5353

54-
// loss = loss / Scalar<double>::make(num_samples);
54+
loss = loss / Scalar<double>::make(num_samples);
5555
loss->backward();
5656

5757
// update weights

0 commit comments

Comments
 (0)