Move Jenkins test to Github action (#7342)

2026-06-29 03:54:16 +08:00 · 2020-04-09 10:27:19 -07:00
parent 061043229f
commit 59867dad75
12 changed files with 259 additions and 118 deletions
@@ -203,7 +203,8 @@ if __name__ == "__main__":
        })

    training_start = time.time()
-    for i in range(3):
+    num_epochs = 1 if args.smoke_test else 3
+    for i in range(num_epochs):
        # Trains num epochs
        train_stats1 = trainer.train()
        train_stats1.update(trainer.validate())