Commit 796a4353 authored by Yuxin Wu's avatar Yuxin Wu

fix import (fix #461)

parent 00473316
...@@ -9,7 +9,7 @@ from tqdm import tqdm ...@@ -9,7 +9,7 @@ from tqdm import tqdm
from six.moves import queue from six.moves import queue
from tensorpack.utils.concurrency import StoppableThread, ShareSessionThread from tensorpack.utils.concurrency import StoppableThread, ShareSessionThread
from tensorpack.callbacks import Triggerable from tensorpack.callbacks import Callback
from tensorpack.utils import logger from tensorpack.utils import logger
from tensorpack.utils.stats import StatCounter from tensorpack.utils.stats import StatCounter
from tensorpack.utils.utils import get_tqdm_kwargs from tensorpack.utils.utils import get_tqdm_kwargs
...@@ -108,7 +108,7 @@ def eval_model_multithread(pred, nr_eval, get_player_fn): ...@@ -108,7 +108,7 @@ def eval_model_multithread(pred, nr_eval, get_player_fn):
logger.info("Average Score: {}; Max Score: {}".format(mean, max)) logger.info("Average Score: {}; Max Score: {}".format(mean, max))
class Evaluator(Triggerable): class Evaluator(Callback):
def __init__(self, nr_eval, input_names, output_names, get_player_fn): def __init__(self, nr_eval, input_names, output_names, get_player_fn):
self.eval_episode = nr_eval self.eval_episode = nr_eval
self.input_names = input_names self.input_names = input_names
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment