提交 422ebb8a authored 作者: Frederic Bastien's avatar Frederic Bastien

Print the warning only once

上级 6ec43120
...@@ -18,6 +18,7 @@ import sys ...@@ -18,6 +18,7 @@ import sys
import time import time
from collections import defaultdict from collections import defaultdict
from six import iteritems from six import iteritems
import warnings
import numpy as np import numpy as np
...@@ -278,10 +279,10 @@ class ProfileStats(object): ...@@ -278,10 +279,10 @@ class ProfileStats(object):
hasattr(theano, 'gpuarray') and hasattr(theano, 'gpuarray') and
theano.gpuarray.pygpu_activated and theano.gpuarray.pygpu_activated and
not config.profiling.ignore_first_call): not config.profiling.ignore_first_call):
logger.warn( warnings.warn(
"Theano flag profiling.ignore_first_call is False." "Theano flag profiling.ignore_first_call is False."
" This cause bad profiling result in the new gpu" " This cause bad profiling result in the new gpu"
" back-end, we as sometimes we compile at the first call.") " back-end, as sometimes we compile at the first call.")
self.apply_callcount = {} self.apply_callcount = {}
self.output_size = {} self.output_size = {}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论