parent
8adb5f4a32
commit
b47fa5f50e
|
@ -44,14 +44,14 @@ class Engine(object):
|
|||
self.mode = mode
|
||||
self.config = config
|
||||
|
||||
# set seed
|
||||
self._init_seed()
|
||||
|
||||
# init logger
|
||||
log_file = os.path.join(self.config['Global']['output_dir'],
|
||||
self.config["Arch"]["name"], f"{mode}.log")
|
||||
init_logger(log_file=log_file)
|
||||
|
||||
# set seed
|
||||
self._init_seed()
|
||||
|
||||
# set device
|
||||
self._init_device()
|
||||
|
||||
|
|
Loading…
Reference in New Issue