我正在尝试用tf.Dataset作为我的数据集来拟合Keras模型。我指定了参数steps_per_epoch。input to a model, you should specify the 'steps_per_epoch' argument.此错误使我感到困惑,因为我将steps_per_epoch参数指定为数据集的长度我已经尝试了None以及小于我的数据集长度的整数,但都没有用。下面是我<
使用Tensorflow的新Dataset API进行多GPU培训(来自TFRecords格式)似乎要比在单个GPU上运行(1对4 Tesla K80s)慢得多(1/4慢)。从nvidia-smi的输出来看,使用4个GPU只会使GPU的利用率达到15%左右,而单个GPU的利用率大约为45%。
从磁盘(tfrecords-格式)加载数据是否会导致培训速度的瓶颈?使用常规feed-dicts,将整个数据集加载到