|
|
@@ -32,7 +32,6 @@ class EvaluationHook(Hook): |
|
|
def do_evaluate(self, trainer): |
|
|
def do_evaluate(self, trainer): |
|
|
"""Evaluate the results.""" |
|
|
"""Evaluate the results.""" |
|
|
eval_res = trainer.evaluate() |
|
|
eval_res = trainer.evaluate() |
|
|
trainer.data_loader = trainer.train_dataloader |
|
|
|
|
|
for name, val in eval_res.items(): |
|
|
for name, val in eval_res.items(): |
|
|
trainer.log_buffer.output[name] = val |
|
|
trainer.log_buffer.output[name] = val |
|
|
|
|
|
|
|
|
|