wenlongli10
wenlongli10
我看了一下，没有错，你可以debug下试试
> @kill2013110 Batch size =4 is too small to stable training. Lowering the learning rate is helpful. > > You could try my another YOLO project: > > https://github.com/yjh0410/PyTorch_YOLO-Family >...
@yjh0410 In other words, do you think there is room for improvement in inference speed based on pytorch？
@yjh0410 ok，thanks
> 请问你解决这个问题了么？ 解决了，因为把loss当做自定义的lambda层也放进网络里了，这样训练时网络的输出就是loss，比较简洁，你可以仔细看训练模型的代码，但这个lambda层是不支持savedmodel格式的。 解决办法：可以建一个不包括自定义loss层的模型，然后将训练好的weight load进去，再savedmodel保存
训练时 ValueError: slice index -1 of dimension 0 out of bounds. for 'loss/yolo_loss_loss/strided_slice'
解决了，真是tf版本问题，升级到tf2.3.0就好了
master分支可以用了吗
目前我在r0.2版本上用yolox-tiny是可以进行图模式和动态图模式训练的
> 上面的报错看起来是pycharm的debug功能相关的报错，可以试下直接在terminal窗口里面用命令行执行试一下 是的是的，没注意看到😂
optimize 'per_batch_map' 这个是个啥参数