add stub for get_training_results within HFT Model
@@ -51,7 +51,8 @@ def list_field(default=None, metadata=None):
# Used to reduce training time when debugging
DEBUG = True
QUICK_TRAIN_BUILD_ARGUMENTS = {
- "max_train_samples": "2",
+ #"max_train_samples": "2", # Setting this causes errors (probably shouldn't)
+ "num_train_epochs": "0.02",
}
# Training Stages
@@ -567,6 +568,10 @@ class HFTransformersModel(BaseModel):
index = TRAINING_STAGES.index(stage)
self.stage = self.index_prefixed_stages[index]
+ def get_train_results(self) -> Dict[str: float]:
+ # TODO Ask Ben what's meant to go here
+ return { "comparison_val": 6.9 }
+
class ElpisTokenizer(Wav2Vec2CTCTokenizer):
"""
