Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于训练的问题 #39

Open
dingyan1478 opened this issue Nov 27, 2021 · 2 comments
Open

关于训练的问题 #39

dingyan1478 opened this issue Nov 27, 2021 · 2 comments

Comments

@dingyan1478
Copy link

您好!
感谢您的工作,我有一个疑问,如果我不使用多进程进行训练的话,我应该如何设置这条命令呢“python -m torch.distributed.launch --nproc_per_node=8 --master_port=4321 basicsr/t.py -opt options/train/GoPro/HINet.yml --launcher pytorch”?
现在我设置的一直在报错,非常期待能够得到您的回复!

@zxcvads
Copy link

zxcvads commented Nov 30, 2021

may be you can try "python basicsr/train.py -opt options/train/GoPro/HINet.yml"

@mayorx
Copy link
Collaborator

mayorx commented Dec 2, 2021

dingyan1478, 您好,
您可以尝试 nono0822的方案, 或者尝试将--nproc_per_node 的值设置为1, https://github.com/megvii-model/HINet/blob/main/options/train/GoPro/HINet.yml#L11 将num_gpu也设置为1

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants