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

请问是否解决了呢?我也遇到了相同的问题。 #19

Open
zhangyonle opened this issue Aug 5, 2024 · 6 comments
Open

请问是否解决了呢?我也遇到了相同的问题。 #19

zhangyonle opened this issue Aug 5, 2024 · 6 comments

Comments

@zhangyonle
Copy link

          请问是否解决了呢?我也遇到了相同的问题。

muten_0_pred

Originally posted by @panda7777777 in #11 (comment)

I operate step by step following readme to test rough_sketches, but I encounter the same question with all simplication sketches is blank white. I can find the what reason leading to this.

@MarkMoHR
Copy link
Owner

MarkMoHR commented Aug 5, 2024

You could try double checking if the pre-trained model weights are placed and loaded correctly.

@panda7777777
Copy link

我记得我当时是这样解决的:1,使用CPU推理而非GPU(在我的服务器上,GPU推理并不会更快,而且当时有一些奇怪的警告信息);2,降低 matplotlib 版本。

@panda7777777
Copy link

另外训练一直是正常的,我可以使用GPU在自己的数据集上训练,并且取得了很好的效果

@zhangyonle
Copy link
Author

我记得我当时是这样解决的:1,使用CPU推理而非GPU(在我的服务器上,GPU推理并不会更快,而且当时有一些奇怪的警告信息);2,降低 matplotlib 版本。

十分感谢,我也用CPU推理就能成功测试出结果了。但是好奇,为啥GPU推理结果就是空白的,我认为应该不是作者说的预训练模型加载不正确的原因。

@zhangyonle
Copy link
Author

You could try double checking if the pre-trained model weights are placed and loaded correctly.

I have tested successfully via CPU inference, Thank panda 7777777. However, once I change to GPU inference only, the testing results are also blank. I do think that pre-trained model weights are not loaded correctly.

@MarkMoHR
Copy link
Owner

MarkMoHR commented Aug 6, 2024

那可能是显卡的问题。我那时候是用1080Ti跑的,可能Tensorflow1.x的代码跟现在新的显卡不兼容,可以尝试用旧一点的显卡

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