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

Fix tensor_meta update bug #5924

Merged
merged 3 commits into from
Aug 18, 2021
Merged

Fix tensor_meta update bug #5924

merged 3 commits into from
Aug 18, 2021

Conversation

wyg1997
Copy link
Contributor

@wyg1997 wyg1997 commented Aug 17, 2021

solve #5889

@github-actions
Copy link
Contributor

Speed stats:
GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 135.8ms (= 6787.8ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 126.2ms (= 6308.5ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.08 (= 135.8ms / 126.2ms)

PyTorch resnet50 time: 83.8ms (= 4190.6ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 72.8ms (= 3641.7ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.15 (= 83.8ms / 72.8ms)

PyTorch resnet50 time: 55.3ms (= 2767.3ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 46.3ms (= 2314.1ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.20 (= 55.3ms / 46.3ms)

PyTorch resnet50 time: 49.4ms (= 2468.0ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 38.2ms (= 1911.2ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.29 (= 49.4ms / 38.2ms)

PyTorch resnet50 time: 44.8ms (= 2239.5ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 39.4ms (= 1971.8ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 1.14 (= 44.8ms / 39.4ms)

@oneflow-ci-bot oneflow-ci-bot merged commit e9915ec into master Aug 18, 2021
@oneflow-ci-bot oneflow-ci-bot deleted the fix-autograd_meta_update branch August 18, 2021 04:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants