Skip to content

Commit

Permalink
try to figure out why tests are taking so long
Browse files Browse the repository at this point in the history
  • Loading branch information
misko committed Oct 4, 2024
1 parent 1e4b909 commit 1ac3aa2
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions tests/core/e2e/test_s2ef.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,14 +336,14 @@ def test_train_and_predict(
tutorial_val_src,
):
# test with amp
self.smoke_test_train(
input_yaml=configs[model_name],
tutorial_val_src=tutorial_val_src,
otf_norms=otf_norms,
world_size=1,
num_workers=0,
amp=True,
)
#self.smoke_test_train(
# input_yaml=configs[model_name],
# tutorial_val_src=tutorial_val_src,
# otf_norms=otf_norms,
# world_size=1,
# num_workers=0,
# amp=True,
#)
# test without amp
self.smoke_test_train(
input_yaml=configs[model_name],
Expand Down

0 comments on commit 1ac3aa2

Please sign in to comment.