Sign upLogin

tensorflow/models

View on GitHub
Star
official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py

Summary

Maintainability
C
1 day
Test Coverage

File mobilebert_edgetpu_trainer.py has 415 lines of code (exceeds 250 allowed). Consider refactoring.
Open

# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
Severity: Minor
Found in official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py - About 5 hrs to fix

    Function train_loop_begin has a Cognitive Complexity of 11 (exceeds 5 allowed). Consider refactoring.
    Open

      def train_loop_begin(self):
    """A train loop is similar with the concept of an epoch."""
    self.train_metrics['feature_transfer_mse'].reset_states()
    self.train_metrics['beta_transfer_loss'].reset_states()
    self.train_metrics['gamma_transfer_loss'].reset_states()
    Severity: Minor
    Found in official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py - About 1 hr to fix

    Cognitive Complexity

    Cognitive Complexity is a measure of how difficult a unit of code is to intuitively understand. Unlike Cyclomatic Complexity, which determines how difficult your code will be to test, Cognitive Complexity tells you how difficult your code will be to read and comprehend.

    A method's cognitive complexity is based on a few simple rules:

    • Code is not considered more complex when it uses shorthand that the language provides for collapsing multiple statements into one
    • Code is considered more complex for each "break in the linear flow of the code"
    • Code is considered more complex when "flow breaking structures are nested"

    Further reading

    Function __init__ has 29 lines of code (exceeds 25 allowed). Consider refactoring.
    Open

      def __init__(self,
               teacher_model: modeling.models.BertPretrainerV2,
               student_model: modeling.models.BertPretrainerV2,
               strategy: tf.distribute.Strategy,
               experiment_params: params.EdgeTPUBERTCustomParams,
    Severity: Minor
    Found in official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py - About 1 hr to fix

      Function __init__ has 6 arguments (exceeds 4 allowed). Consider refactoring.
      Open

        def __init__(self,
      Severity: Minor
      Found in official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py - About 45 mins to fix

        There are no issues that match your filters.

        Category
        Status