TF2 使用TPU Compilation failure: Dynamic dimension propagation on reversed dimension is not supported
在tensorflow2下使用TPU训练时
...
tf.pad(inputs, [[0, 0], [10, 10], [10, 10], [0, 0]], mode='reflect')
...
batch_size = 1 * tpu_strategy.num_replicas_in_sync
data_set = tf.data.Dataset.from_tensor_slices(X_data) # X_data is a numpy Array
data_set.batch(ba...
原创
2020-05-26 12:46:39 ·
462 阅读 ·
1 评论