Merge pull request #9 from rafid-dev/validation

Luecx · web-flow · commit 7ee831f8acd4 · 2024-03-29T19:38:01.000+01:00
Fix typo real quick
diff --git a/src/main.cu b/src/main.cu
@@ -23,7 +23,7 @@ int main(int argc, char* argv[]) {
         .help("Total positions in each epoch")
         .scan<'i', int>();
 
-    program.add_argument("--val--size")
+    program.add_argument("--val-size")
         .default_value(10000000)
         .help("Total positions for each validation epoch")
         .scan<'i', int>();
@@ -117,6 +117,7 @@ int main(int argc, char* argv[]) {
 
     std::cout << "Epochs: " << total_epochs << "\n"
               << "Epochs Size: " << epoch_size << "\n"
+              << "Validation Size: " << val_epoch_size << "\n"
               << "Save Rate: " << save_rate << "\n"
               << "FT Size: " << ft_size << "\n"
               << "Lambda: " << lambda << "\n"
