search

bubbliiiing / centernet-tf2

这是一个centernet-tf2的源码，可以用于训练自己的模型。
MIT License
46 stars 27 forks source link

train.py报错 提示维度索引过界 #2

Open TingsongYu opened 3 years ago

TingsongYu commented 3 years ago

楼主你好，在复现train.py时，出现这个报错，请问是哪里不匹配呢？ “”ValueError: slice index -1 of dimension 0 out of bounds. for 'loss/centernet_loss_loss/strided_slice' (op: 'StridedSlice') with input shapes: [0], [1], [1], [1] and with computed input tensors: input[1] = <-1>, input[2] = <0>, input[3] = <1>.” Traceback (most recent call last): File "F:/github_desktop/centernet-tf2-main/train.py", line 214, in optimizer=keras.optimizers.Adam(Lr) 期待解答，感谢!

bubbliiiing commented 3 years ago

应该是版本问题的

bubbliiiing commented 3 years ago

tf2.2