search

caiyuanhao1998 / Retinexformer

"Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement" (ICCV 2023) & (NTIRE 2024 Challenge)
https://arxiv.org/abs/2303.06705
MIT License
828 stars 64 forks source link

分布式训练问题 #74

Closed 348330 closed 4 months ago

348330 commented 4 months ago

image 作者您好，我想请教一下模型如何进行分布式训练？以及如果我有两个GPU，分别是GPU0和GPU1，如何设置模型只在GPU1上运行?谢谢！

caiyuanhao1998 commented 4 months ago

请按照我们的 readme 里面的介绍配置环境并执行命令进行分布式训练。

如果觉得我们的 repo 有用的话，帮忙点点 star 支持一下