:zap: Add a timer

app.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import time
 from typing import Any
 
 from huggingface_hub import PyTorchModelHubMixin
@@ -426,9 +427,19 @@ def start_bert(classifier_model, model_save_path, criterion, WINDOW, batch_size=
 
 
 if __name__ == "__main__":
+  start_time = time.time()
+
   dataset_folder_prefix = "inputdata/"
   pytorch_model = MQtlDnaBERT6Classifier()
   start_bert(classifier_model=pytorch_model, model_save_path=f"weights_{pytorch_model.model_name}.pth",
              criterion=ReshapedBCEWithLogitsLoss(), WINDOW=4000, batch_size=12,  # max 14 on my laptop...
              dataset_folder_prefix=dataset_folder_prefix, max_epochs=1)
+
+  # Record the end time
+  end_time = time.time()
+  # Calculate the duration
+  duration = end_time - start_time
+  # Print the runtime
+  print(f"Runtime: {duration:.2f} seconds")
+
   pass