Remove some obsolete HF Transformers test cases (#942)

GoogleCloudPlatform · Aug 9, 2023 · 434c27c · 434c27c
1 parent f596aea
commit 434c27c
Showing 1 changed file with 0 additions and 89 deletions.
diff --git a/tests/pytorch/nightly/hf-glue.libsonnet b/tests/pytorch/nightly/hf-glue.libsonnet
@@ -30,33 +30,6 @@ local utils = import 'templates/utils.libsonnet';
   local command_copy_metrics = |||
     gsutil -m cp -r ./tensorboard-metrics/* $(MODEL_DIR)
   |||,
-  local bert_base_cased = self.bert_base_cased,
-  bert_base_cased:: common.Convergence {
-    modelName: 'hf-glue-bert-b-c',
-    paramsOverride+:: {
-      model_name_or_path: 'bert-base-cased',
-      per_device_train_batch_size: 64,
-      per_device_eval_batch_size: 64,
-    },
-    metricConfig+: {
-      sourceMap+:: {
-        tensorboard+: {
-          aggregateAssertionsMap+:: {
-            'eval/accuracy': {
-              FINAL: {
-                fixed_value: {
-                  comparison: 'GREATER',
-                  value: 0.80,
-                },
-                inclusive_bounds: false,
-                wait_for_n_data_points: 0,
-              },
-            },
-          },
-        },
-      },
-    },
-  },
   local hf_glue = self.hf_glue,
   hf_glue:: common.PyTorchTest {
     local config = self,
@@ -114,60 +87,6 @@ local utils = import 'templates/utils.libsonnet';
       },
     },
   },
-  local xlnet_large_cased = self.xlnet_large_cased,
-  xlnet_large_cased:: common.Convergence {
-    modelName: 'hf-glue-xlnet-l-c',
-    paramsOverride+:: {
-      model_name_or_path: 'xlnet-large-cased',
-      per_device_train_batch_size: 32,
-      per_device_eval_batch_size: 16,
-    },
-    metricConfig+: {
-      sourceMap+:: {
-        tensorboard+: {
-          aggregateAssertionsMap+:: {
-            'eval/accuracy': {
-              FINAL: {
-                fixed_value: {
-                  comparison: 'GREATER',
-                  value: 0.85,
-                },
-                inclusive_bounds: false,
-                wait_for_n_data_points: 0,
-              },
-            },
-          },
-        },
-      },
-    },
-  },
-  local roberta_large = self.roberta_large,
-  roberta_large:: common.Convergence {
-    modelName: 'hf-glue-roberta-l',
-    paramsOverride+:: {
-      model_name_or_path: 'roberta-large',
-      per_device_train_batch_size: 16,
-      per_device_eval_batch_size: 16,
-    },
-    metricConfig+: {
-      sourceMap+:: {
-        tensorboard+: {
-          aggregateAssertionsMap+:: {
-            'eval/accuracy': {
-              FINAL: {
-                fixed_value: {
-                  comparison: 'GREATER',
-                  value: 0.85,
-                },
-                inclusive_bounds: false,
-                wait_for_n_data_points: 0,
-              },
-            },
-          },
-        },
-      },
-    },
-  },
   local distilbert_base_uncased = self.distilbert_base_uncased,
   distilbert_base_uncased:: common.Convergence {
     modelName: 'hf-glue-distilbert-b-uc',
@@ -209,10 +128,6 @@ local utils = import 'templates/utils.libsonnet';
   pjrt:: common.PyTorchTpuVmMixin + tpuVm {
     modelName: 'hf-glue-pjrt',
   },
-  local v2_8 = self.v2_8,
-  v2_8:: {
-    accelerator: tpus.v2_8,
-  },
   local v3_8 = self.v3_8,
   v3_8:: {
     accelerator: tpus.v3_8,
@@ -222,11 +137,7 @@ local utils = import 'templates/utils.libsonnet';
     accelerator: tpus.v4_8,
   },
   configs: [
-    hf_glue + v2_8 + bert_base_cased + timeouts.Hours(2) + xrt,
-    hf_glue + v3_8 + xlnet_large_cased + timeouts.Hours(5) + xrt,
-    hf_glue + v3_8 + roberta_large + timeouts.Hours(8) + xrt,
     hf_glue + v3_8 + distilbert_base_uncased + timeouts.Hours(2) + xrt,
-    hf_glue + v4_8 + roberta_large + timeouts.Hours(6) + xrt,
     hf_glue + v4_8 + distilbert_base_uncased + timeouts.Hours(2) + pjrt,
   ],
 }