Commit d18344bf authored by Paul Bethge's avatar Paul Bethge
Browse files

allow mem growth

parent 0d2b75ca
......@@ -28,8 +28,8 @@ class AudioAugmenter(object):
self._aug_flow = flow.Sequential([
shift,
crop,
vltp,
speed,
#vltp,
#speed,
pitch,
noise,
])
......
......@@ -134,7 +134,8 @@ if __name__ == "__main__":
help="Path to the required config file.")
cli_args = parser.parse_args()
tf.config.list_physical_devices('GPU')
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)
# copy models & config for later
log_dir = os.path.join("logs", datetime.now().strftime("%Y-%m-%d-%H-%M-%S"))
......
Supports Markdown
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