Skip to content

Commit

Permalink
Merge pull request #730 from mv1388/improve_gpu_test_naming
Browse files Browse the repository at this point in the history
Make GPU test names more descriptive and consistent
  • Loading branch information
mv1388 committed Aug 8, 2022
2 parents 00826c6 + 455cfe3 commit d58deee
Show file tree
Hide file tree
Showing 8 changed files with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ def get_predictions(self, batch_data, device):
return predictions.cpu(), batch["labels"].cpu(), {}


class TestIMDBBERT(unittest.TestCase):
class TestAMPIMDBBERT(unittest.TestCase):
def test_trainloop_core_pytorch_compare(self):
train_data, test_data = self.get_data_sets(ds_subset_size=1000)

Expand Down

0 comments on commit d58deee

Please sign in to comment.