text-summarization-tensorflow icon indicating copy to clipboard operation
text-summarization-tensorflow copied to clipboard

Training times

Open gogasca opened this issue 7 years ago • 4 comments

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)

gogasca avatar Oct 10 '18 08:10 gogasca

I am wondering the same.

TheRockXu avatar Oct 19 '18 06:10 TheRockXu

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:

vvssttkk avatar Nov 12 '18 11:11 vvssttkk

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 avatar Jul 11 '19 10:07 adenaufal

@adenaufal where u fit or predict your nn or at the head at jupyter

vvssttkk avatar Jul 11 '19 10:07 vvssttkk