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

请问运行train.py时卡住了怎么处理? #94

Open
oxxj opened this issue Sep 2, 2024 · 4 comments
Open

请问运行train.py时卡住了怎么处理? #94

oxxj opened this issue Sep 2, 2024 · 4 comments

Comments

@oxxj
Copy link

oxxj commented Sep 2, 2024

我在运行python train.py -c configs/train_propainter.json时,显示下面这种情况

(propainter) ➜ ProPainter git:(main) ✗ python train.py -c configs/train_propainter.json
world_size: 4
using GPU 0-0 for training
[**] create folder experiments_model/propainter_train_propainter
using GPU 1-1 for training
using GPU 2-2 for training
using GPU 3-3 for training
Pretrained flow completion model has loaded...
Pretrained flow completion model has loaded...
Network [InpaintGenerator] was created. Total number of parameters: 39.4 million. To see the architecture, do print(network).
Pretrained flow completion model has loaded...
Network [InpaintGenerator] was created. Total number of parameters: 39.4 million. To see the architecture, do print(network).
Pretrained flow completion model has loaded...
Network [InpaintGenerator] was created. Total number of parameters: 39.4 million. To see the architecture, do print(network).
Network [InpaintGenerator] was created. Total number of parameters: 39.4 million. To see the architecture, do print(network).
Warnning: There is no trained model found.An initialized model will be used.
0%| | 0/700000 [00:00<?, ?it/s]

另外运行python train.py -c configs/train_flowcomp.json时也一直停在0%这儿,请问这个怎么处理?

@nxrcqupt01
Copy link

我也是这种情况, 请问你解决了吗

@oxxj
Copy link
Author

oxxj commented Oct 12, 2024

我也是这种情况,请问解决了吗

我修改了model/propainter.py的183行,将x.view修改为x.reshape

@shenyewei
Copy link

请问你的训练时长是多少?我在4张A100跑,需要1000个小时,这正常吗?

@oxxj
Copy link
Author

oxxj commented Oct 31, 2024

请问你的训练时长是多少?我在4张A100跑,需要1000个小时,这正常吗?

我当时也跑了很久,用的4张A6000,如果你只是想快速跑完可以修改下train_flowcomp.json/train_propainter.json里的"iterations"参数看看

@oxxj oxxj closed this as completed Oct 31, 2024
@oxxj oxxj reopened this Oct 31, 2024
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