Skip to content

Commit

Permalink
correct __doc__
Browse files Browse the repository at this point in the history
  • Loading branch information
linjing-lab committed Oct 25, 2023
1 parent c3bd3ef commit f860b4a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions released_box/perming/_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,7 +206,7 @@ def train_val(self,
:param interval: int, console output interval. default: 100.
:param tolerance: float, tolerance set to judge difference in val_loss. default: 1e-3
:param patience: int, patience of no improvement waiting for training to stop. default: 10.
:param backend: str, 'threading', 'multiprocessing', 'locky'. default: 'threading'.
:param backend: str, 'threading', 'multiprocessing', 'loky'. default: 'threading'.
:param n_jobs: int, accelerate processing of validation. default: -1.
:param early_stop: bool, whether to enable early_stop in train_val. default: False.
'''
Expand Down Expand Up @@ -354,7 +354,7 @@ def _pack_info(self, by: str, state: bool) -> Dict[str, Any]:
def _set_container(self, backend: str, n_jobs: int) -> None:
'''
Validation Container with `parallel_backend` at `n_jobs`.
:param backend: str, "threading", "multiprocessing, 'locky'.
:param backend: str, "threading", "multiprocessing, 'loky'.
:param n_jobs: int, set jobs with backend to accelerate process.
'''
with parallel_backend(backend, n_jobs=n_jobs):
Expand Down

0 comments on commit f860b4a

Please sign in to comment.