Commit 100ae5a0 authored by Yuxin Wu's avatar Yuxin Wu

warn about #497

parent 48375921
...@@ -24,8 +24,9 @@ To train, just run: ...@@ -24,8 +24,9 @@ To train, just run:
```bash ```bash
./imagenet-resnet.py --data /path/to/original/ILSVRC --gpu 0,1,2,3 -d 50 [--mode resnet/preact/se] ./imagenet-resnet.py --data /path/to/original/ILSVRC --gpu 0,1,2,3 -d 50 [--mode resnet/preact/se]
``` ```
You should be able to see good GPU utilization (around 95%), if your data is fast enough. You should be able to see good GPU utilization (95%~99%), if your data is fast enough.
See the [tutorial](http://tensorpack.readthedocs.io/en/latest/tutorial/efficient-dataflow.html) on how to speed up your data. The default data pipeline is probably OK for most systems.
See the [tutorial](http://tensorpack.readthedocs.io/en/latest/tutorial/efficient-dataflow.html) on other options to speed up your data.
![imagenet](imagenet-resnet.png) ![imagenet](imagenet-resnet.png)
......
...@@ -5,6 +5,7 @@ ...@@ -5,6 +5,7 @@
import tensorflow as tf import tensorflow as tf
from .common import get_default_sess_config from .common import get_default_sess_config
from ..utils import logger
__all__ = ['NewSessionCreator', 'ReuseSessionCreator', 'SessionCreatorAdapter'] __all__ = ['NewSessionCreator', 'ReuseSessionCreator', 'SessionCreatorAdapter']
...@@ -33,6 +34,9 @@ class NewSessionCreator(tf.train.ChiefSessionCreator): ...@@ -33,6 +34,9 @@ class NewSessionCreator(tf.train.ChiefSessionCreator):
config = get_default_sess_config() config = get_default_sess_config()
else: else:
self.user_provided_config = True self.user_provided_config = True
logger.warn(
"Some options in custom session config may not work due to TF \
bugs. See https://github.com/ppwwyyxx/tensorpack/issues/497 for workarounds.")
self.config = config self.config = config
super(NewSessionCreator, self).__init__(master=target, config=config) super(NewSessionCreator, self).__init__(master=target, config=config)
......
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