region loss存在错误

[linklist2]

value = -torch.topk(-gray, int(gray.shape[0] * 0.4))[0][0]

您好，您这里求得了gray的最小值，而不是40%所在的值。 举例如下图所示：

我觉得应该是这样写可能更符合论文中描述的前40%最暗的像素值：

[rxqH]

您好，我想问一个如果是单GPU，应该如何修改训练代码

[rebornone1]

@rxqH 你好你解决单GPU训练问题了吗 谢谢 我的qq 1185718748