benjaminStreltzin commited on
Commit
a81ac51
·
verified ·
1 Parent(s): b8e4555

Update vit_Training.py

Browse files
Files changed (1) hide show
  1. vit_Training.py +2 -2
vit_Training.py CHANGED
@@ -77,7 +77,7 @@ class Custom_VIT_Model:
77
  # Print the current state of the training data for debugging
78
  print("Current training data:")
79
  print(self.df)
80
- print("urod")
81
 
82
 
83
 
@@ -85,7 +85,7 @@ class Custom_VIT_Model:
85
  if len(self.df) >= 100:
86
  print("Retraining the model as we have enough data.")
87
  self.retrain_model()
88
- self.df = None
89
 
90
 
91
 
 
77
  # Print the current state of the training data for debugging
78
  print("Current training data:")
79
  print(self.df)
80
+
81
 
82
 
83
 
 
85
  if len(self.df) >= 100:
86
  print("Retraining the model as we have enough data.")
87
  self.retrain_model()
88
+
89
 
90
 
91