Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Remove translation benchmark
Browse files Browse the repository at this point in the history
  • Loading branch information
ethanwharris committed Feb 15, 2022
1 parent 0d88ad6 commit 9d07113
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions flash/text/seq2seq/translation/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import Any, Dict, List, Optional, Union
from typing import Any, Dict, Optional, Union

from torchmetrics import BLEUScore

Expand Down Expand Up @@ -99,8 +99,3 @@ def compute_metrics(self, generated_tokens, batch, prefix):
else:
result = self.bleu(reference_corpus, translate_corpus)
self.log(f"{prefix}_bleu_score", result, on_step=False, on_epoch=True, prog_bar=True)

@staticmethod
def _ci_benchmark_fn(history: List[Dict[str, Any]]):
"""This function is used only for debugging usage with CI."""
assert history[-1]["val_bleu_score"] > 0.6, history[-1]["val_bleu_score"]

0 comments on commit 9d07113

Please sign in to comment.