Update docs (#71)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
e629335f6d
commit
d85b44f259
11 changed files with 286 additions and 35 deletions
|
|
@ -35,8 +35,8 @@ RANK = int(os.getenv('RANK', -1))
|
|||
|
||||
class BaseTrainer:
|
||||
|
||||
def __init__(self, config=DEFAULT_CONFIG, overrides={}):
|
||||
self.args = get_config(config, overrides)
|
||||
def __init__(self, cfg=DEFAULT_CONFIG, overrides={}):
|
||||
self.args = get_config(cfg, overrides)
|
||||
self.check_resume()
|
||||
init_seeds(self.args.seed + 1 + RANK, deterministic=self.args.deterministic)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue