Compare commits
1 Commits
fix/eval-a
...
chore/docs
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
159f0531f9 |
@@ -215,6 +215,10 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
|
|||||||
self.cfg.auto_find_batch_size
|
self.cfg.auto_find_batch_size
|
||||||
)
|
)
|
||||||
|
|
||||||
|
training_arguments_kwargs["eval_accumulation_steps"] = (
|
||||||
|
self.cfg.gradient_accumulation_steps
|
||||||
|
)
|
||||||
|
|
||||||
training_arguments_kwargs["load_best_model_at_end"] = (
|
training_arguments_kwargs["load_best_model_at_end"] = (
|
||||||
(
|
(
|
||||||
self.cfg.load_best_model_at_end is not False
|
self.cfg.load_best_model_at_end is not False
|
||||||
|
|||||||
@@ -74,9 +74,6 @@ def is_main_process() -> bool:
|
|||||||
Check if the current process is the main process. If not in distributed mode,
|
Check if the current process is the main process. If not in distributed mode,
|
||||||
always return `True`.
|
always return `True`.
|
||||||
|
|
||||||
We use a simpler logic when the distributed state is not initialized: we just log
|
|
||||||
on the 0-th local rank.
|
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
`True` if the current process is the main process, `False` otherwise.
|
`True` if the current process is the main process, `False` otherwise.
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user