|
@@ -408,7 +408,7 @@ class UnitYFinetune:
|
|
def run(self) -> None:
|
|
def run(self) -> None:
|
|
logger.info("Start Finetuning")
|
|
logger.info("Start Finetuning")
|
|
self._reset_stats()
|
|
self._reset_stats()
|
|
- self._eval_model()
|
|
|
|
|
|
+ self._eval_model(n_batches=100)
|
|
|
|
|
|
train_dataloader = self.train_data_loader.get_dataloader()
|
|
train_dataloader = self.train_data_loader.get_dataloader()
|
|
|
|
|