config = tf.ConfigProto() config.gpu_options.allow_growth = True sess = tf.Session(graph=graph, config=config)
沒有留言:
張貼留言