Commit b1850e0b authored by Yuxin Wu's avatar Yuxin Wu

fix old examples again

parent 0cfc0889
...@@ -117,6 +117,7 @@ class Trainer(object): ...@@ -117,6 +117,7 @@ class Trainer(object):
logger.warn("This could happen if you wrote a custom trainer before.") logger.warn("This could happen if you wrote a custom trainer before.")
logger.warn("It may work now through some hacks, but please switch to the new API!") logger.warn("It may work now through some hacks, but please switch to the new API!")
logger.warn("See https://github.com/ppwwyyxx/tensorpack/issues/458 for more information.") logger.warn("See https://github.com/ppwwyyxx/tensorpack/issues/458 for more information.")
config._deprecated_parsing()
self._config = config self._config = config
self.inputs_desc = config.model.get_inputs_desc() self.inputs_desc = config.model.get_inputs_desc()
self.tower_func = TowerFuncWrapper( self.tower_func = TowerFuncWrapper(
......
...@@ -151,6 +151,8 @@ class TrainConfig(object): ...@@ -151,6 +151,8 @@ class TrainConfig(object):
self.nr_tower = kwargs.pop('nr_tower') self.nr_tower = kwargs.pop('nr_tower')
if 'tower' in kwargs: if 'tower' in kwargs:
self.tower = kwargs.pop('tower') self.tower = kwargs.pop('tower')
else:
self.tower = [0]
assert len(kwargs) == 0, "Unknown arguments: {}".format(kwargs.keys()) assert len(kwargs) == 0, "Unknown arguments: {}".format(kwargs.keys())
@property @property
......
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