From d87a68737c11ebf45ce96046e6371f11d9c76c2e Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Fri, 29 Nov 2024 23:16:41 -0500 Subject: [PATCH] new algorithm --- tests/test_full_nlp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_full_nlp.py b/tests/test_full_nlp.py index 8a4145a99f..5785ebb096 100644 --- a/tests/test_full_nlp.py +++ b/tests/test_full_nlp.py @@ -11,7 +11,7 @@ from torchmetrics.classification import MulticlassAccuracy from transformers import BertConfig, BertForMaskedLM, BertForSequenceClassification, BertTokenizerFast -from composer.algorithms import LayerFreezing +from composer.algorithms import GradientClipping from composer.loggers import RemoteUploaderDownloader from composer.metrics.nlp import LanguageCrossEntropy, MaskedAccuracy from composer.models import HuggingFaceModel @@ -233,7 +233,7 @@ def inference_test_helper( @pytest.mark.parametrize( 'model_type,algorithms,save_format', [ - ('tinybert_hf', [LayerFreezing], 'onnx'), + ('tinybert_hf', [GradientClipping], 'onnx'), ('simpletransformer', [], 'torchscript'), ], )