format

gradsflow · Nov 11, 2021 · 1c203bd · 1c203bd
1 parent de2efb0
commit 1c203bd
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/gradsflow/core/base.py b/gradsflow/core/base.py
@@ -60,7 +60,6 @@ def reset_metrics(self):
 class BaseTracker:
     max_epochs: int = 0
     current_epoch: int = 0  # current train current_epoch
-    # current_step: int = 0  # current current_step
     steps_per_epoch: Optional[int] = None
     train: TrackingValues = TrackingValues()
     val: TrackingValues = TrackingValues()
diff --git a/gradsflow/models/tracker.py b/gradsflow/models/tracker.py
@@ -11,7 +11,6 @@
 #  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.
-import warnings
 from typing import Dict, List
 
 from rich import box
@@ -42,7 +41,6 @@ def mode(self, mode) -> TrackingValues:
     def track(self, key, value):
         """Tracks value"""
         epoch = self.current_epoch
-        # step = self.current_step
         data = {"current_epoch": epoch, key: to_item(value)}
         self.logs.append(data)
 
@@ -101,7 +99,6 @@ def create_table(self) -> Table:
     def reset(self):
         self.max_epochs = 0
         self.current_epoch = 0
-        # self.current_step = 0  # deprecated
         self.steps_per_epoch = None
         self.train = TrackingValues()
         self.val = TrackingValues()