速度太慢了

[wangwuwu]

我用Converse2d替换yolo-seg proto的上采样，训练时间直接变成4倍了，请问下作者的训练速度怎么样

[csgeekhuang]

As the current code has not been specifically optimized for speed, the slower performance is reasonable.

[wangwuwu]

作者有优化的计划吗

[Yiozolm]

我实现了简单的CUDA kernel，不知道你有没有兴趣尝试一下Converse2d_CUDA

[Starsm7]

@Yiozolm Good!

[ouyangxi0925]

我用Converse2d替换yolo-seg proto的上采样，训练时间直接变成4倍了，请问下作者的训练速度怎么样

您好，我使用converse2d替换yolo的upsample上采样，出现了 File "/home/lab917/anaconda3/envs/oldxi-v8/lib/python3.8/multiprocessing/connection.py", line 379, in _recv
chunk = read(handle, remaining)
ConnectionResetError: [Errno 104] Connection reset by peer
请问您有遇到过吗？
您的cuda和torch版本是多少呢？

[Yiozolm]

我用Converse2d替换yolo-seg proto的上采样，训练时间直接变成4倍了，请问下作者的训练速度怎么样

您好，我使用converse2d替换yolo的upsample上采样，出现了 File "/home/lab917/anaconda3/envs/oldxi-v8/lib/python3.8/multiprocessing/connection.py", line 379, in _recv chunk = read(handle, remaining) ConnectionResetError: [Errno 104] Connection reset by peer 请问您有遇到过吗？ 您的cuda和torch版本是多少呢？

我自己的下游任务里没遇到过这种情况，是不是因为python版本差太多了。

• Python 3.12
• CUDA 12.4
• Torch 2.4.0

[So-cean]

Hello，i am trying to use Converse2D to replace the Upsample function in cyclegan, but it's extremely slow. Almost 10x slower.