Skip to content

Commit 346884c

Browse files
committed
Revert "Revert "Merge branch 'master' into thick_conv_lstm""
This reverts commit 7339103.
1 parent a28f42c commit 346884c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

train.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ def _parse_function(example):
202202

203203
if not weight_data:
204204
weights = tf.ones(tf.shape(weights))
205+
else:
206+
weights = tf.tile(tf.reduce_mean(weights, axis=0, keep_dims=True), [tf.shape(cameras)[0],])
205207

206208

207209
# return features and labels

0 commit comments

Comments
 (0)