testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #47 from iclementine/reborn 

fix ExperimentBase
This commit is contained in:
Feiyu Chan 2020-12-11 12:04:26 +08:00 committed by GitHub
parent 07ce84c680 cecc8735c4
commit a5f1605051
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
parakeet/training/experiment.py
View File

@ -109,7 +109,7 @@ class ExperimentBase(object):
def train(self):
self.new_epoch()
while self.iteration <= self.config.training.max_iteration:
while self.iteration < self.config.training.max_iteration:
self.iteration += 1
self.train_batch()
@ -131,7 +131,7 @@ class ExperimentBase(object):
def setup_output_dir(self):
# output dir
output_dir = Path(self.args.output).expanduser()
output_dir.mkdir(exist_ok=True)
output_dir.mkdir(parents=True, exist_ok=True)
self.output_dir = output_dir