text-summarization-tensorflow
text-summarization-tensorflow copied to clipboard
Training times
Anyone care to comment training times for this model? CPU/GPU/TPU? I'm thinking in creating a Google Cloud ML Engine package for this (For both training and serving)
I am wondering the same.
I think u can run the code on GPU or TPU, I tried to use GPU & the time decreased twice to do this u shoud write next code:
config = tf.ConfigProto(device_count = {'GPU': 1})
with tf.Session(config=config) as sess:
I think u can run the code on GPU or TPU, I tried to use GPU & the time decreased twice to do this u shoud write next code:
config = tf.ConfigProto(device_count = {'GPU': 1}) with tf.Session(config=config) as sess:
@vtrokhymenko in which file do you put the code?
@adenaufal where u fit or predict your nn or at the head at jupyter