mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix typro 

close #268
pull/299/head
liaoxingyu 2020-09-14 11:34:28 +08:00
parent cf246e9bee
commit 4fa3f08a4a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fastreid/engine/defaults.py
View File

@ -346,7 +346,7 @@ class DefaultTrainer(SimpleTrainer):
return [
# It may not always print what you want to see, since it prints "common" metrics only.
CommonMetricPrinter(self.max_iter),
JSONWriter(os.path.join(self.cfg.OUTPUT_DIR, "metrics.json")),
# JSONWriter(os.path.join(self.cfg.OUTPUT_DIR, "metrics.json")),
TensorboardXWriter(self.cfg.OUTPUT_DIR),
]
@ -429,7 +429,7 @@ class DefaultTrainer(SimpleTrainer):
model (nn.Module):
evaluators (list[DatasetEvaluator] or None): if None, will call
:meth:`build_evaluator`. Otherwise, must have the same length as
`cfg.DATASETS.TEST`.
`cfg.DATASETS.TESTS`.
Returns:
dict: a dict of result metrics
"""
@ -438,8 +438,8 @@ class DefaultTrainer(SimpleTrainer):
evaluators = [evaluators]
if evaluators is not None:
assert len(cfg.DATASETS.TEST) == len(evaluators), "{} != {}".format(
len(cfg.DATASETS.TEST), len(evaluators)
assert len(cfg.DATASETS.TESTS) == len(evaluators), "{} != {}".format(
len(cfg.DATASETS.TESTS), len(evaluators)
)
results = OrderedDict()