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

Update nigthly whl version #10528

Merged
merged 1 commit into from
May 22, 2024
Merged

Update nigthly whl version #10528

merged 1 commit into from
May 22, 2024

Conversation

mosout
Copy link
Contributor

@mosout mosout commented May 21, 2024

No description provided.

Copy link
Contributor

Copy link
Contributor

Speed stats:
GPU Name: NVIDIA GeForce RTX 3080 Ti 

❌ OneFlow resnet50 time: 43.4ms (= 4337.9ms / 100, input_shape=[16, 3, 224, 224])
PyTorch resnet50 time: 57.5ms (= 5745.8ms / 100, input_shape=[16, 3, 224, 224])
✔️ Relative speed: 1.32 (= 57.5ms / 43.4ms)

OneFlow resnet50 time: 26.4ms (= 2643.5ms / 100, input_shape=[8, 3, 224, 224])
PyTorch resnet50 time: 37.7ms (= 3768.4ms / 100, input_shape=[8, 3, 224, 224])
✔️ Relative speed: 1.43 (= 37.7ms / 26.4ms)

OneFlow resnet50 time: 18.3ms (= 3660.7ms / 200, input_shape=[4, 3, 224, 224])
PyTorch resnet50 time: 36.6ms (= 7325.5ms / 200, input_shape=[4, 3, 224, 224])
✔️ Relative speed: 2.00 (= 36.6ms / 18.3ms)

OneFlow resnet50 time: 17.0ms (= 3409.5ms / 200, input_shape=[2, 3, 224, 224])
PyTorch resnet50 time: 31.4ms (= 6274.4ms / 200, input_shape=[2, 3, 224, 224])
✔️ Relative speed: 1.84 (= 31.4ms / 17.0ms)

OneFlow resnet50 time: 16.5ms (= 3299.0ms / 200, input_shape=[1, 3, 224, 224])
PyTorch resnet50 time: 31.6ms (= 6313.5ms / 200, input_shape=[1, 3, 224, 224])
✔️ Relative speed: 1.91 (= 31.6ms / 16.5ms)

OneFlow swin dataloader time: 0.201s (= 40.169s / 200, num_workers=1)
PyTorch swin dataloader time: 0.128s (= 25.584s / 200, num_workers=1)
Relative speed: 0.637 (= 0.128s / 0.201s)

OneFlow swin dataloader time: 0.054s (= 10.803s / 200, num_workers=4)
PyTorch swin dataloader time: 0.033s (= 6.529s / 200, num_workers=4)
Relative speed: 0.604 (= 0.033s / 0.054s)

OneFlow swin dataloader time: 0.029s (= 5.743s / 200, num_workers=8)
PyTorch swin dataloader time: 0.016s (= 3.257s / 200, num_workers=8)
Relative speed: 0.567 (= 0.016s / 0.029s)

❌ OneFlow resnet50 time: 49.2ms (= 4920.9ms / 100, input_shape=[16, 3, 224, 224], ddp, world size=2)
PyTorch resnet50 time: 65.7ms (= 6572.8ms / 100, input_shape=[16, 3, 224, 224], ddp, world size=2)
✔️ Relative speed: 1.34 (= 65.7ms / 49.2ms)

OneFlow resnet50 time: 36.7ms (= 3670.7ms / 100, input_shape=[8, 3, 224, 224], ddp, world size=2)
PyTorch resnet50 time: 47.8ms (= 4783.3ms / 100, input_shape=[8, 3, 224, 224], ddp, world size=2)
✔️ Relative speed: 1.30 (= 47.8ms / 36.7ms)

OneFlow resnet50 time: 27.7ms (= 5548.2ms / 200, input_shape=[4, 3, 224, 224], ddp, world size=2)
PyTorch resnet50 time: 42.6ms (= 8513.5ms / 200, input_shape=[4, 3, 224, 224], ddp, world size=2)
✔️ Relative speed: 1.53 (= 42.6ms / 27.7ms)

OneFlow resnet50 time: 25.1ms (= 5025.5ms / 200, input_shape=[2, 3, 224, 224], ddp, world size=2)
PyTorch resnet50 time: 39.0ms (= 7808.5ms / 200, input_shape=[2, 3, 224, 224], ddp, world size=2)
✔️ Relative speed: 1.55 (= 39.0ms / 25.1ms)

OneFlow resnet50 time: 24.8ms (= 4958.2ms / 200, input_shape=[1, 3, 224, 224], ddp, world size=2)
PyTorch resnet50 time: 35.9ms (= 7183.7ms / 200, input_shape=[1, 3, 224, 224], ddp, world size=2)
✔️ Relative speed: 1.45 (= 35.9ms / 24.8ms)

@mosout mosout merged commit 29f9434 into master May 22, 2024
34 of 41 checks passed
@mosout mosout deleted the nightly_whl_version branch May 22, 2024 02:15
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.

None yet

2 participants