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
Add speed test script #117
Merged
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Ldpe2G
requested review from
rentainhe,
kaijieshi7,
daquexian and
oneflow-ci-bot
and removed request for
rentainhe
January 21, 2022 05:12
之前有个想法,既然有返回所有models的方法,那么能不能直接对返回的模型进行测试,这样就不需要每个模型都需要单独的脚本。存在的问题是,每个模型除了名字不一样之外,还存在input数据增强不一样,输出也不一样(googlenet输出三个)。 |
因为需要和torch的对比速度,所以不是仅仅从 model_creater 拿到模型就行了,那只能测oneflow的模型速度,要测torch的速度得把模型文件开头的 import oneflow 改成 import torch,需要修改一下脚本代码,然后保存下来再重新加载测试torch的速度 |
…ow-Inc/vision into dev_add_speed_test_scripts
以下的没问题值能跑通。
|
好的,我再看看 |
Ldpe2G
requested review from
oneflow-ci-bot
and removed request for
oneflow-ci-bot
January 22, 2022 15:48
Ldpe2G
requested review from
oneflow-ci-bot
and removed request for
oneflow-ci-bot
January 23, 2022 04:52
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
脚本运行方式:
cd ci/check bash run_speed_test.sh
结果会输出到 当前目录下的
result
文件里目前通过测速脚本发现的问题
import torch as flow
运行会崩本身运行也会报错,输入是
224x224
的时候