-
Notifications
You must be signed in to change notification settings - Fork 350
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
[Versioning] Versions for 0.6 #2509
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
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/2509
Note: Links to docs will display an error until the docs builds have been completed. ❌ 2 New Failures, 4 Unrelated FailuresAs of commit 0fe3b56 with merge base 7ff03a2 ( NEW FAILURES - The following jobs have failed:
BROKEN TRUNK - The following jobs failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
8ee0c71
to
6306ae8
Compare
|
Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
---|---|---|---|---|---|
test_simple | 0.7313s | 0.7307s | 1.3686 Ops/s | 1.3292 Ops/s | |
test_transformed | 1.0622s | 0.9862s | 1.0140 Ops/s | 1.0033 Ops/s | |
test_serial | 2.2089s | 2.1366s | 0.4680 Ops/s | 0.4561 Ops/s | |
test_parallel | 2.0715s | 1.9956s | 0.5011 Ops/s | 0.4906 Ops/s | |
test_step_mdp_speed[True-True-True-True-True] | 0.1497ms | 39.5233μs | 25.3015 KOps/s | 24.8768 KOps/s | |
test_step_mdp_speed[True-True-True-True-False] | 56.1020μs | 23.2204μs | 43.0656 KOps/s | 42.8923 KOps/s | |
test_step_mdp_speed[True-True-True-False-True] | 61.8330μs | 21.4608μs | 46.5966 KOps/s | 45.8515 KOps/s | |
test_step_mdp_speed[True-True-True-False-False] | 53.2230μs | 12.6887μs | 78.8100 KOps/s | 80.4959 KOps/s | |
test_step_mdp_speed[True-True-False-True-True] | 89.7540μs | 42.5893μs | 23.4801 KOps/s | 22.9592 KOps/s | |
test_step_mdp_speed[True-True-False-True-False] | 59.1030μs | 25.9210μs | 38.5787 KOps/s | 38.5707 KOps/s | |
test_step_mdp_speed[True-True-False-False-True] | 63.2530μs | 24.6140μs | 40.6274 KOps/s | 40.3368 KOps/s | |
test_step_mdp_speed[True-True-False-False-False] | 53.3320μs | 15.5077μs | 64.4840 KOps/s | 65.2941 KOps/s | |
test_step_mdp_speed[True-False-True-True-True] | 0.1033ms | 44.5498μs | 22.4468 KOps/s | 22.0180 KOps/s | |
test_step_mdp_speed[True-False-True-True-False] | 61.1630μs | 28.7070μs | 34.8348 KOps/s | 34.6986 KOps/s | |
test_step_mdp_speed[True-False-True-False-True] | 56.0820μs | 24.2653μs | 41.2111 KOps/s | 40.2140 KOps/s | |
test_step_mdp_speed[True-False-True-False-False] | 54.4520μs | 15.5999μs | 64.1031 KOps/s | 65.0293 KOps/s | |
test_step_mdp_speed[True-False-False-True-True] | 94.0850μs | 48.5989μs | 20.5766 KOps/s | 20.5987 KOps/s | |
test_step_mdp_speed[True-False-False-True-False] | 67.1130μs | 31.0265μs | 32.2305 KOps/s | 32.0425 KOps/s | |
test_step_mdp_speed[True-False-False-False-True] | 64.9230μs | 26.7635μs | 37.3643 KOps/s | 36.9163 KOps/s | |
test_step_mdp_speed[True-False-False-False-False] | 48.0920μs | 18.0789μs | 55.3130 KOps/s | 55.5900 KOps/s | |
test_step_mdp_speed[False-True-True-True-True] | 80.9930μs | 45.0704μs | 22.1875 KOps/s | 21.8034 KOps/s | |
test_step_mdp_speed[False-True-True-True-False] | 61.7830μs | 28.9377μs | 34.5570 KOps/s | 35.1027 KOps/s | |
test_step_mdp_speed[False-True-True-False-True] | 57.5520μs | 29.4724μs | 33.9301 KOps/s | 33.9856 KOps/s | |
test_step_mdp_speed[False-True-True-False-False] | 52.7120μs | 17.9489μs | 55.7136 KOps/s | 55.4558 KOps/s | |
test_step_mdp_speed[False-True-False-True-True] | 83.4540μs | 48.5402μs | 20.6015 KOps/s | 20.5766 KOps/s | |
test_step_mdp_speed[False-True-False-True-False] | 63.6130μs | 31.8074μs | 31.4393 KOps/s | 32.0865 KOps/s | |
test_step_mdp_speed[False-True-False-False-True] | 3.0872ms | 32.0304μs | 31.2204 KOps/s | 31.3489 KOps/s | |
test_step_mdp_speed[False-True-False-False-False] | 55.2730μs | 20.7174μs | 48.2685 KOps/s | 51.3362 KOps/s | |
test_step_mdp_speed[False-False-True-True-True] | 92.7750μs | 51.2942μs | 19.4954 KOps/s | 19.9774 KOps/s | |
test_step_mdp_speed[False-False-True-True-False] | 67.8530μs | 34.1912μs | 29.2473 KOps/s | 29.5985 KOps/s | |
test_step_mdp_speed[False-False-True-False-True] | 64.5430μs | 32.0507μs | 31.2006 KOps/s | 31.6392 KOps/s | |
test_step_mdp_speed[False-False-True-False-False] | 51.3520μs | 20.7101μs | 48.2856 KOps/s | 49.5174 KOps/s | |
test_step_mdp_speed[False-False-False-True-True] | 83.7040μs | 53.1466μs | 18.8159 KOps/s | 18.9270 KOps/s | |
test_step_mdp_speed[False-False-False-True-False] | 69.8730μs | 36.9099μs | 27.0930 KOps/s | 27.2214 KOps/s | |
test_step_mdp_speed[False-False-False-False-True] | 64.1830μs | 33.1148μs | 30.1980 KOps/s | 29.1830 KOps/s | |
test_step_mdp_speed[False-False-False-False-False] | 58.2320μs | 22.9672μs | 43.5404 KOps/s | 43.0894 KOps/s | |
test_values[generalized_advantage_estimate-True-True] | 26.0496ms | 24.8603ms | 40.2248 Ops/s | 41.2742 Ops/s | |
test_values[vec_generalized_advantage_estimate-True-True] | 98.4611ms | 2.8559ms | 350.1474 Ops/s | 352.5499 Ops/s | |
test_values[td0_return_estimate-False-False] | 86.4940μs | 65.1553μs | 15.3479 KOps/s | 15.4330 KOps/s | |
test_values[td1_return_estimate-False-False] | 55.3549ms | 54.5566ms | 18.3296 Ops/s | 18.4756 Ops/s | |
test_values[vec_td1_return_estimate-False-False] | 1.3693ms | 1.0724ms | 932.5039 Ops/s | 930.4293 Ops/s | |
test_values[td_lambda_return_estimate-True-False] | 90.2899ms | 86.4389ms | 11.5689 Ops/s | 11.5170 Ops/s | |
test_values[vec_td_lambda_return_estimate-True-False] | 1.3654ms | 1.0657ms | 938.3739 Ops/s | 936.0993 Ops/s | |
test_gae_speed[generalized_advantage_estimate-False-1-512] | 25.5785ms | 24.3430ms | 41.0796 Ops/s | 40.9628 Ops/s | |
test_gae_speed[vec_generalized_advantage_estimate-True-1-512] | 1.0309ms | 0.7455ms | 1.3414 KOps/s | 1.3355 KOps/s | |
test_gae_speed[vec_generalized_advantage_estimate-False-1-512] | 0.7525ms | 0.6578ms | 1.5203 KOps/s | 1.5187 KOps/s | |
test_gae_speed[vec_generalized_advantage_estimate-True-32-512] | 1.5083ms | 1.4688ms | 680.8247 Ops/s | 680.2461 Ops/s | |
test_gae_speed[vec_generalized_advantage_estimate-False-32-512] | 0.7946ms | 0.6854ms | 1.4591 KOps/s | 1.4722 KOps/s | |
test_dqn_speed[False-None] | 6.5003ms | 1.3468ms | 742.5100 Ops/s | 723.7291 Ops/s | |
test_dqn_speed[False-backward] | 1.9116ms | 1.8301ms | 546.4168 Ops/s | 536.9202 Ops/s | |
test_dqn_speed[True-None] | 0.7206ms | 0.5865ms | 1.7051 KOps/s | 1.6711 KOps/s | |
test_dqn_speed[True-backward] | 1.1136ms | 1.0572ms | 945.8924 Ops/s | 846.2824 Ops/s | |
test_dqn_speed[reduce-overhead-None] | 0.7173ms | 0.6027ms | 1.6591 KOps/s | 1.7273 KOps/s | |
test_dqn_speed[reduce-overhead-backward] | 1.1681ms | 1.0376ms | 963.8052 Ops/s | 970.9620 Ops/s | |
test_ddpg_speed[False-None] | 3.2188ms | 2.7630ms | 361.9275 Ops/s | 360.7569 Ops/s | |
test_ddpg_speed[False-backward] | 4.4456ms | 4.0153ms | 249.0484 Ops/s | 247.6010 Ops/s | |
test_ddpg_speed[True-None] | 1.4847ms | 1.3061ms | 765.6195 Ops/s | 786.8414 Ops/s | |
test_ddpg_speed[True-backward] | 2.3662ms | 2.2823ms | 438.1510 Ops/s | 369.2192 Ops/s | |
test_ddpg_speed[reduce-overhead-None] | 1.6728ms | 1.2947ms | 772.3784 Ops/s | 781.3464 Ops/s | |
test_ddpg_speed[reduce-overhead-backward] | 2.3791ms | 2.2645ms | 441.5982 Ops/s | 373.4914 Ops/s | |
test_sac_speed[False-None] | 8.2791ms | 7.7321ms | 129.3302 Ops/s | 127.7051 Ops/s | |
test_sac_speed[False-backward] | 12.0901ms | 11.0022ms | 90.8909 Ops/s | 90.9476 Ops/s | |
test_sac_speed[True-None] | 2.2282ms | 2.0700ms | 483.0859 Ops/s | 477.5029 Ops/s | |
test_sac_speed[True-backward] | 4.1467ms | 4.0059ms | 249.6324 Ops/s | 239.1956 Ops/s | |
test_sac_speed[reduce-overhead-None] | 2.1715ms | 2.0598ms | 485.4839 Ops/s | 479.2367 Ops/s | |
test_sac_speed[reduce-overhead-backward] | 4.2283ms | 4.0562ms | 246.5362 Ops/s | 249.6839 Ops/s | |
test_redq_speed[False-None] | 15.4700ms | 10.6673ms | 93.7447 Ops/s | 91.4898 Ops/s | |
test_redq_speed[False-backward] | 18.6099ms | 17.4458ms | 57.3202 Ops/s | 57.1470 Ops/s | |
test_redq_speed[True-None] | 3.8427ms | 3.4510ms | 289.7724 Ops/s | 270.1626 Ops/s | |
test_redq_speed[True-backward] | 8.5721ms | 8.3600ms | 119.6178 Ops/s | 115.9923 Ops/s | |
test_redq_speed[reduce-overhead-None] | 3.7355ms | 3.3582ms | 297.7767 Ops/s | 276.7437 Ops/s | |
test_redq_speed[reduce-overhead-backward] | 8.5428ms | 8.2567ms | 121.1141 Ops/s | 117.0010 Ops/s | |
test_redq_deprec_speed[False-None] | 11.2426ms | 10.7411ms | 93.1002 Ops/s | 92.4460 Ops/s | |
test_redq_deprec_speed[False-backward] | 16.0678ms | 15.6694ms | 63.8185 Ops/s | 64.6458 Ops/s | |
test_redq_deprec_speed[True-None] | 3.4919ms | 3.3446ms | 298.9853 Ops/s | 304.0588 Ops/s | |
test_redq_deprec_speed[True-backward] | 7.9026ms | 7.4020ms | 135.0992 Ops/s | 137.7552 Ops/s | |
test_redq_deprec_speed[reduce-overhead-None] | 3.4347ms | 3.3000ms | 303.0264 Ops/s | 307.1086 Ops/s | |
test_redq_deprec_speed[reduce-overhead-backward] | 7.5426ms | 7.3885ms | 135.3464 Ops/s | 141.9481 Ops/s | |
test_td3_speed[False-None] | 7.7241ms | 7.5418ms | 132.5938 Ops/s | 127.8506 Ops/s | |
test_td3_speed[False-backward] | 10.9178ms | 10.3967ms | 96.1846 Ops/s | 92.2489 Ops/s | |
test_td3_speed[True-None] | 1.9849ms | 1.9450ms | 514.1469 Ops/s | 512.7374 Ops/s | |
test_td3_speed[True-backward] | 3.8975ms | 3.7860ms | 264.1295 Ops/s | 223.7916 Ops/s | |
test_td3_speed[reduce-overhead-None] | 2.0167ms | 1.9578ms | 510.7649 Ops/s | 502.3660 Ops/s | |
test_td3_speed[reduce-overhead-backward] | 3.8625ms | 3.7582ms | 266.0843 Ops/s | 262.0493 Ops/s | |
test_cql_speed[False-None] | 27.9992ms | 25.2029ms | 39.6780 Ops/s | 39.4586 Ops/s | |
test_cql_speed[False-backward] | 38.1552ms | 34.6846ms | 28.8312 Ops/s | 28.1902 Ops/s | |
test_cql_speed[True-None] | 11.1423ms | 10.7683ms | 92.8651 Ops/s | 92.3704 Ops/s | |
test_cql_speed[True-backward] | 16.8270ms | 16.5315ms | 60.4906 Ops/s | 58.3853 Ops/s | |
test_cql_speed[reduce-overhead-None] | 11.2240ms | 10.8384ms | 92.2647 Ops/s | 91.0340 Ops/s | |
test_cql_speed[reduce-overhead-backward] | 16.8948ms | 16.5764ms | 60.3267 Ops/s | 59.3543 Ops/s | |
test_a2c_speed[False-None] | 5.8281ms | 5.3210ms | 187.9341 Ops/s | 183.5122 Ops/s | |
test_a2c_speed[False-backward] | 12.6584ms | 12.0645ms | 82.8875 Ops/s | 83.8379 Ops/s | |
test_a2c_speed[True-None] | 3.3220ms | 3.0564ms | 327.1855 Ops/s | 325.5699 Ops/s | |
test_a2c_speed[True-backward] | 8.7580ms | 8.5973ms | 116.3150 Ops/s | 110.8601 Ops/s | |
test_a2c_speed[reduce-overhead-None] | 3.5461ms | 3.1058ms | 321.9754 Ops/s | 322.8685 Ops/s | |
test_a2c_speed[reduce-overhead-backward] | 8.8475ms | 8.6421ms | 115.7126 Ops/s | 117.4725 Ops/s | |
test_ppo_speed[False-None] | 6.1014ms | 5.7253ms | 174.6632 Ops/s | 176.3316 Ops/s | |
test_ppo_speed[False-backward] | 12.7662ms | 12.4435ms | 80.3635 Ops/s | 81.1184 Ops/s | |
test_ppo_speed[True-None] | 3.8200ms | 3.4749ms | 287.7749 Ops/s | 275.2265 Ops/s | |
test_ppo_speed[True-backward] | 9.0679ms | 8.5161ms | 117.4250 Ops/s | 120.0270 Ops/s | |
test_ppo_speed[reduce-overhead-None] | 3.8759ms | 3.4691ms | 288.2562 Ops/s | 286.1785 Ops/s | |
test_ppo_speed[reduce-overhead-backward] | 8.7774ms | 8.3314ms | 120.0282 Ops/s | 118.8834 Ops/s | |
test_reinforce_speed[False-None] | 4.8593ms | 4.4903ms | 222.7006 Ops/s | 217.8953 Ops/s | |
test_reinforce_speed[False-backward] | 7.5525ms | 7.3280ms | 136.4634 Ops/s | 134.7723 Ops/s | |
test_reinforce_speed[True-None] | 2.6401ms | 2.2340ms | 447.6303 Ops/s | 432.5472 Ops/s | |
test_reinforce_speed[True-backward] | 7.5850ms | 7.2155ms | 138.5901 Ops/s | 139.5555 Ops/s | |
test_reinforce_speed[reduce-overhead-None] | 2.7230ms | 2.2370ms | 447.0318 Ops/s | 440.5030 Ops/s | |
test_reinforce_speed[reduce-overhead-backward] | 7.3320ms | 7.1602ms | 139.6611 Ops/s | 140.3401 Ops/s | |
test_iql_speed[False-None] | 19.7091ms | 19.2880ms | 51.8458 Ops/s | 51.2349 Ops/s | |
test_iql_speed[False-backward] | 30.4594ms | 29.8418ms | 33.5100 Ops/s | 33.4287 Ops/s | |
test_iql_speed[True-None] | 7.1700ms | 6.7478ms | 148.1963 Ops/s | 141.0297 Ops/s | |
test_iql_speed[True-backward] | 16.1851ms | 15.6644ms | 63.8392 Ops/s | 63.7641 Ops/s | |
test_iql_speed[reduce-overhead-None] | 7.1780ms | 6.7556ms | 148.0245 Ops/s | 145.7791 Ops/s | |
test_iql_speed[reduce-overhead-backward] | 15.9435ms | 15.6449ms | 63.9187 Ops/s | 63.6198 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-ListStorage-RandomSampler-4000] | 6.7629ms | 6.3648ms | 157.1130 Ops/s | 157.1695 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-LazyMemmapStorage-RandomSampler-10000] | 0.9671ms | 0.3344ms | 2.9907 KOps/s | 4.1713 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-LazyTensorStorage-RandomSampler-10000] | 0.8522ms | 0.3177ms | 3.1476 KOps/s | 4.5836 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-4000] | 6.6891ms | 6.1600ms | 162.3366 Ops/s | 165.5313 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-LazyMemmapStorage-SamplerWithoutReplacement-10000] | 1.8580ms | 0.3168ms | 3.1569 KOps/s | 4.2224 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-LazyTensorStorage-SamplerWithoutReplacement-10000] | 0.7448ms | 0.3070ms | 3.2574 KOps/s | 4.6527 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-LazyMemmapStorage-sampler6-10000] | 1.6884ms | 1.3491ms | 741.2344 Ops/s | 812.3584 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-LazyTensorStorage-sampler7-10000] | 1.5490ms | 1.2882ms | 776.2650 Ops/s | 844.7005 Ops/s | |
test_rb_sample[TensorDictPrioritizedReplayBuffer-ListStorage-None-4000] | 6.7756ms | 6.3131ms | 158.4019 Ops/s | 161.1522 Ops/s | |
test_rb_sample[TensorDictPrioritizedReplayBuffer-LazyMemmapStorage-None-10000] | 2.4894ms | 0.4576ms | 2.1854 KOps/s | 2.0992 KOps/s | |
test_rb_sample[TensorDictPrioritizedReplayBuffer-LazyTensorStorage-None-10000] | 0.9761ms | 0.4538ms | 2.2035 KOps/s | 2.1675 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-ListStorage-RandomSampler-4000] | 6.6251ms | 6.1869ms | 161.6319 Ops/s | 164.6401 Ops/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyMemmapStorage-RandomSampler-10000] | 1.6312ms | 0.2601ms | 3.8451 KOps/s | 4.0953 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyTensorStorage-RandomSampler-10000] | 0.5878ms | 0.3383ms | 2.9556 KOps/s | 4.4594 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-4000] | 13.2759ms | 6.3475ms | 157.5425 Ops/s | 165.9433 Ops/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyMemmapStorage-SamplerWithoutReplacement-10000] | 0.8663ms | 0.2380ms | 4.2014 KOps/s | 4.1692 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyTensorStorage-SamplerWithoutReplacement-10000] | 0.5173ms | 0.3497ms | 2.8595 KOps/s | 4.5995 KOps/s | |
test_rb_iterate[TensorDictPrioritizedReplayBuffer-ListStorage-None-4000] | 6.4188ms | 6.2850ms | 159.1096 Ops/s | 161.9005 Ops/s | |
test_rb_iterate[TensorDictPrioritizedReplayBuffer-LazyMemmapStorage-None-10000] | 2.5447ms | 0.4550ms | 2.1980 KOps/s | 649.7164 Ops/s | |
test_rb_iterate[TensorDictPrioritizedReplayBuffer-LazyTensorStorage-None-10000] | 0.5959ms | 0.4249ms | 2.3535 KOps/s | 2.6982 KOps/s | |
test_rb_populate[TensorDictReplayBuffer-ListStorage-RandomSampler-400] | 6.8393ms | 5.2621ms | 190.0373 Ops/s | 182.5709 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyMemmapStorage-RandomSampler-400] | 7.7328ms | 2.0249ms | 493.8431 Ops/s | 482.2205 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyTensorStorage-RandomSampler-400] | 8.2857ms | 1.2220ms | 818.3082 Ops/s | 819.6630 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-400] | 0.4008s | 13.2772ms | 75.3173 Ops/s | 186.4218 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyMemmapStorage-SamplerWithoutReplacement-400] | 8.4185ms | 2.0224ms | 494.4700 Ops/s | 489.4621 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyTensorStorage-SamplerWithoutReplacement-400] | 7.0338ms | 1.2561ms | 796.1362 Ops/s | 803.3769 Ops/s | |
test_rb_populate[TensorDictPrioritizedReplayBuffer-ListStorage-None-400] | 7.1041ms | 5.5395ms | 180.5225 Ops/s | 179.7725 Ops/s | |
test_rb_populate[TensorDictPrioritizedReplayBuffer-LazyMemmapStorage-None-400] | 10.5254ms | 2.2528ms | 443.8973 Ops/s | 419.7672 Ops/s | |
test_rb_populate[TensorDictPrioritizedReplayBuffer-LazyTensorStorage-None-400] | 7.1596ms | 1.3986ms | 715.0069 Ops/s | 663.5332 Ops/s |
This was referenced Oct 22, 2024
6306ae8
to
ca359e2
Compare
|
Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
---|---|---|---|---|---|
test_simple | 0.4132s | 0.4112s | 2.4320 Ops/s | 2.3221 Ops/s | |
test_transformed | 0.6810s | 0.6050s | 1.6528 Ops/s | 1.6522 Ops/s | |
test_serial | 1.4180s | 1.3414s | 0.7455 Ops/s | 0.7476 Ops/s | |
test_parallel | 1.3699s | 1.3137s | 0.7612 Ops/s | 0.7575 Ops/s | |
test_step_mdp_speed[True-True-True-True-True] | 0.1738ms | 28.8425μs | 34.6710 KOps/s | 34.5198 KOps/s | |
test_step_mdp_speed[True-True-True-True-False] | 54.7930μs | 17.1606μs | 58.2730 KOps/s | 57.0514 KOps/s | |
test_step_mdp_speed[True-True-True-False-True] | 59.1800μs | 15.9649μs | 62.6376 KOps/s | 61.7391 KOps/s | |
test_step_mdp_speed[True-True-True-False-False] | 29.8160μs | 9.4652μs | 105.6506 KOps/s | 104.0214 KOps/s | |
test_step_mdp_speed[True-True-False-True-True] | 84.1670μs | 30.9061μs | 32.3560 KOps/s | 31.2710 KOps/s | |
test_step_mdp_speed[True-True-False-True-False] | 68.4580μs | 19.1080μs | 52.3341 KOps/s | 50.9243 KOps/s | |
test_step_mdp_speed[True-True-False-False-True] | 49.4020μs | 18.2254μs | 54.8685 KOps/s | 54.2078 KOps/s | |
test_step_mdp_speed[True-True-False-False-False] | 51.6370μs | 11.6586μs | 85.7739 KOps/s | 84.7176 KOps/s | |
test_step_mdp_speed[True-False-True-True-True] | 86.3510μs | 32.8951μs | 30.3996 KOps/s | 27.9201 KOps/s | |
test_step_mdp_speed[True-False-True-True-False] | 52.0370μs | 21.3574μs | 46.8222 KOps/s | 45.1335 KOps/s | |
test_step_mdp_speed[True-False-True-False-True] | 45.7650μs | 18.0477μs | 55.4087 KOps/s | 55.0070 KOps/s | |
test_step_mdp_speed[True-False-True-False-False] | 41.1770μs | 11.4406μs | 87.4078 KOps/s | 84.6359 KOps/s | |
test_step_mdp_speed[True-False-False-True-True] | 97.2210μs | 34.8340μs | 28.7076 KOps/s | 28.0263 KOps/s | |
test_step_mdp_speed[True-False-False-True-False] | 57.1370μs | 23.2767μs | 42.9615 KOps/s | 41.8209 KOps/s | |
test_step_mdp_speed[True-False-False-False-True] | 80.6520μs | 19.8275μs | 50.4351 KOps/s | 48.9857 KOps/s | |
test_step_mdp_speed[True-False-False-False-False] | 43.8110μs | 13.6064μs | 73.4951 KOps/s | 74.7428 KOps/s | |
test_step_mdp_speed[False-True-True-True-True] | 76.0410μs | 32.5532μs | 30.7189 KOps/s | 29.6593 KOps/s | |
test_step_mdp_speed[False-True-True-True-False] | 51.4560μs | 21.1581μs | 47.2631 KOps/s | 45.8185 KOps/s | |
test_step_mdp_speed[False-True-True-False-True] | 55.6340μs | 21.5397μs | 46.4260 KOps/s | 46.2961 KOps/s | |
test_step_mdp_speed[False-True-True-False-False] | 48.1890μs | 13.5387μs | 73.8623 KOps/s | 74.5691 KOps/s | |
test_step_mdp_speed[False-True-False-True-True] | 71.4030μs | 35.0519μs | 28.5291 KOps/s | 28.3448 KOps/s | |
test_step_mdp_speed[False-True-False-True-False] | 56.0340μs | 23.5600μs | 42.4449 KOps/s | 41.0523 KOps/s | |
test_step_mdp_speed[False-True-False-False-True] | 2.5121ms | 23.7381μs | 42.1264 KOps/s | 41.9539 KOps/s | |
test_step_mdp_speed[False-True-False-False-False] | 49.6520μs | 15.2838μs | 65.4290 KOps/s | 64.3464 KOps/s | |
test_step_mdp_speed[False-False-True-True-True] | 0.1004ms | 36.9668μs | 27.0513 KOps/s | 26.8137 KOps/s | |
test_step_mdp_speed[False-False-True-True-False] | 66.2130μs | 25.5839μs | 39.0871 KOps/s | 38.4305 KOps/s | |
test_step_mdp_speed[False-False-True-False-True] | 73.6350μs | 22.9972μs | 43.4836 KOps/s | 41.8635 KOps/s | |
test_step_mdp_speed[False-False-True-False-False] | 51.6760μs | 15.1391μs | 66.0543 KOps/s | 64.3565 KOps/s | |
test_step_mdp_speed[False-False-False-True-True] | 88.4450μs | 38.2940μs | 26.1138 KOps/s | 25.0918 KOps/s | |
test_step_mdp_speed[False-False-False-True-False] | 96.7500μs | 27.5727μs | 36.2678 KOps/s | 35.9919 KOps/s | |
test_step_mdp_speed[False-False-False-False-True] | 99.4400μs | 24.6298μs | 40.6013 KOps/s | 39.4820 KOps/s | |
test_step_mdp_speed[False-False-False-False-False] | 64.5600μs | 17.0138μs | 58.7758 KOps/s | 56.8551 KOps/s | |
test_values[generalized_advantage_estimate-True-True] | 9.7247ms | 9.4017ms | 106.3633 Ops/s | 102.6333 Ops/s | |
test_values[vec_generalized_advantage_estimate-True-True] | 36.1481ms | 33.3282ms | 30.0046 Ops/s | 30.0689 Ops/s | |
test_values[td0_return_estimate-False-False] | 0.2304ms | 0.1698ms | 5.8902 KOps/s | 5.9524 KOps/s | |
test_values[td1_return_estimate-False-False] | 24.9332ms | 23.9103ms | 41.8230 Ops/s | 40.8014 Ops/s | |
test_values[vec_td1_return_estimate-False-False] | 36.7158ms | 33.1973ms | 30.1229 Ops/s | 29.7355 Ops/s | |
test_values[td_lambda_return_estimate-True-False] | 38.0086ms | 34.0871ms | 29.3366 Ops/s | 28.8748 Ops/s | |
test_values[vec_td_lambda_return_estimate-True-False] | 35.8598ms | 33.2649ms | 30.0617 Ops/s | 29.6995 Ops/s | |
test_gae_speed[generalized_advantage_estimate-False-1-512] | 11.1133ms | 8.3605ms | 119.6097 Ops/s | 116.8549 Ops/s | |
test_gae_speed[vec_generalized_advantage_estimate-True-1-512] | 2.8061ms | 2.3435ms | 426.7169 Ops/s | 548.4821 Ops/s | |
test_gae_speed[vec_generalized_advantage_estimate-False-1-512] | 0.4401ms | 0.3537ms | 2.8271 KOps/s | 2.7858 KOps/s | |
test_gae_speed[vec_generalized_advantage_estimate-True-32-512] | 47.4648ms | 44.2323ms | 22.6079 Ops/s | 23.3540 Ops/s | |
test_gae_speed[vec_generalized_advantage_estimate-False-32-512] | 3.8714ms | 3.0422ms | 328.7104 Ops/s | 326.1959 Ops/s | |
test_dqn_speed[False-None] | 5.5369ms | 1.3576ms | 736.6205 Ops/s | 734.0895 Ops/s | |
test_dqn_speed[False-backward] | 2.9433ms | 1.8693ms | 534.9522 Ops/s | 548.6938 Ops/s | |
test_dqn_speed[True-None] | 0.6025ms | 0.4621ms | 2.1638 KOps/s | 2.1315 KOps/s | |
test_dqn_speed[True-backward] | 1.0566ms | 0.8808ms | 1.1354 KOps/s | 802.1326 Ops/s | |
test_dqn_speed[reduce-overhead-None] | 0.6150ms | 0.4638ms | 2.1563 KOps/s | 2.0691 KOps/s | |
test_dqn_speed[reduce-overhead-backward] | 0.9189ms | 0.8793ms | 1.1373 KOps/s | 1.1275 KOps/s | |
test_ddpg_speed[False-None] | 3.4538ms | 2.8180ms | 354.8651 Ops/s | 354.9943 Ops/s | |
test_ddpg_speed[False-backward] | 5.0479ms | 4.0113ms | 249.2959 Ops/s | 248.1426 Ops/s | |
test_ddpg_speed[True-None] | 1.3305ms | 1.0082ms | 991.8528 Ops/s | 973.8392 Ops/s | |
test_ddpg_speed[True-backward] | 1.9720ms | 1.8925ms | 528.4085 Ops/s | 518.1681 Ops/s | |
test_ddpg_speed[reduce-overhead-None] | 1.3131ms | 1.0123ms | 987.8146 Ops/s | 984.1341 Ops/s | |
test_ddpg_speed[reduce-overhead-backward] | 2.0792ms | 1.9570ms | 510.9862 Ops/s | 525.0123 Ops/s | |
test_sac_speed[False-None] | 10.4712ms | 8.4697ms | 118.0674 Ops/s | 126.1703 Ops/s | |
test_sac_speed[False-backward] | 12.6035ms | 11.4535ms | 87.3092 Ops/s | 92.7244 Ops/s | |
test_sac_speed[True-None] | 2.0635ms | 1.9026ms | 525.6094 Ops/s | 538.5749 Ops/s | |
test_sac_speed[True-backward] | 4.3813ms | 3.8906ms | 257.0281 Ops/s | 282.3997 Ops/s | |
test_sac_speed[reduce-overhead-None] | 2.2273ms | 1.9492ms | 513.0220 Ops/s | 533.1823 Ops/s | |
test_sac_speed[reduce-overhead-backward] | 4.0450ms | 3.5757ms | 279.6670 Ops/s | 283.3385 Ops/s | |
test_redq_speed[False-None] | 15.4419ms | 13.0335ms | 76.7254 Ops/s | 79.4833 Ops/s | |
test_redq_speed[False-backward] | 23.9568ms | 22.1367ms | 45.1740 Ops/s | 45.1937 Ops/s | |
test_redq_speed[True-None] | 5.8527ms | 4.7262ms | 211.5885 Ops/s | 219.8861 Ops/s | |
test_redq_speed[True-backward] | 13.2475ms | 12.0826ms | 82.7638 Ops/s | 80.6972 Ops/s | |
test_redq_speed[reduce-overhead-None] | 5.2809ms | 4.5266ms | 220.9187 Ops/s | 208.2050 Ops/s | |
test_redq_speed[reduce-overhead-backward] | 13.6144ms | 12.1756ms | 82.1312 Ops/s | 83.4009 Ops/s | |
test_redq_deprec_speed[False-None] | 14.6759ms | 12.7774ms | 78.2629 Ops/s | 78.8034 Ops/s | |
test_redq_deprec_speed[False-backward] | 21.2917ms | 18.9844ms | 52.6747 Ops/s | 54.0264 Ops/s | |
test_redq_deprec_speed[True-None] | 4.5867ms | 3.8526ms | 259.5661 Ops/s | 279.3884 Ops/s | |
test_redq_deprec_speed[True-backward] | 9.7748ms | 8.5100ms | 117.5082 Ops/s | 125.1835 Ops/s | |
test_redq_deprec_speed[reduce-overhead-None] | 4.4753ms | 3.8762ms | 257.9865 Ops/s | 279.6999 Ops/s | |
test_redq_deprec_speed[reduce-overhead-backward] | 9.1383ms | 8.7609ms | 114.1435 Ops/s | 126.5176 Ops/s | |
test_td3_speed[False-None] | 8.2674ms | 7.8680ms | 127.0978 Ops/s | 126.5865 Ops/s | |
test_td3_speed[False-backward] | 12.2987ms | 10.3351ms | 96.7577 Ops/s | 96.7065 Ops/s | |
test_td3_speed[True-None] | 1.9910ms | 1.7664ms | 566.1342 Ops/s | 561.5100 Ops/s | |
test_td3_speed[True-backward] | 3.5566ms | 3.4345ms | 291.1633 Ops/s | 293.7616 Ops/s | |
test_td3_speed[reduce-overhead-None] | 1.8689ms | 1.7354ms | 576.2387 Ops/s | 568.6166 Ops/s | |
test_td3_speed[reduce-overhead-backward] | 3.4049ms | 3.3163ms | 301.5439 Ops/s | 294.4823 Ops/s | |
test_cql_speed[False-None] | 37.0854ms | 34.9622ms | 28.6023 Ops/s | 27.5948 Ops/s | |
test_cql_speed[False-backward] | 0.3012s | 50.1941ms | 19.9227 Ops/s | 22.0524 Ops/s | |
test_cql_speed[True-None] | 16.5020ms | 15.4366ms | 64.7809 Ops/s | 64.2797 Ops/s | |
test_cql_speed[True-backward] | 22.7751ms | 21.6239ms | 46.2451 Ops/s | 45.2676 Ops/s | |
test_cql_speed[reduce-overhead-None] | 17.5435ms | 15.5852ms | 64.1633 Ops/s | 63.0701 Ops/s | |
test_cql_speed[reduce-overhead-backward] | 24.1819ms | 22.3814ms | 44.6799 Ops/s | 44.8792 Ops/s | |
test_a2c_speed[False-None] | 8.4867ms | 7.0969ms | 140.9073 Ops/s | 140.6144 Ops/s | |
test_a2c_speed[False-backward] | 15.6241ms | 14.1887ms | 70.4785 Ops/s | 69.7988 Ops/s | |
test_a2c_speed[True-None] | 3.6026ms | 3.3166ms | 301.5129 Ops/s | 298.5270 Ops/s | |
test_a2c_speed[True-backward] | 10.0982ms | 9.6906ms | 103.1923 Ops/s | 101.8688 Ops/s | |
test_a2c_speed[reduce-overhead-None] | 3.6913ms | 3.3107ms | 302.0539 Ops/s | 298.0604 Ops/s | |
test_a2c_speed[reduce-overhead-backward] | 10.0119ms | 9.7044ms | 103.0456 Ops/s | 103.1060 Ops/s | |
test_ppo_speed[False-None] | 8.7894ms | 7.3297ms | 136.4318 Ops/s | 133.6018 Ops/s | |
test_ppo_speed[False-backward] | 16.3359ms | 14.5622ms | 68.6711 Ops/s | 68.4093 Ops/s | |
test_ppo_speed[True-None] | 4.1413ms | 3.6965ms | 270.5226 Ops/s | 266.2704 Ops/s | |
test_ppo_speed[True-backward] | 10.2010ms | 9.5967ms | 104.2021 Ops/s | 103.8591 Ops/s | |
test_ppo_speed[reduce-overhead-None] | 4.4494ms | 3.7113ms | 269.4439 Ops/s | 266.4053 Ops/s | |
test_ppo_speed[reduce-overhead-backward] | 9.9024ms | 9.5518ms | 104.6918 Ops/s | 104.4906 Ops/s | |
test_reinforce_speed[False-None] | 9.6062ms | 6.6349ms | 150.7182 Ops/s | 153.3618 Ops/s | |
test_reinforce_speed[False-backward] | 10.7906ms | 9.9173ms | 100.8340 Ops/s | 102.7456 Ops/s | |
test_reinforce_speed[True-None] | 3.2871ms | 2.6331ms | 379.7749 Ops/s | 375.8748 Ops/s | |
test_reinforce_speed[True-backward] | 9.3064ms | 8.5835ms | 116.5030 Ops/s | 116.2682 Ops/s | |
test_reinforce_speed[reduce-overhead-None] | 3.0717ms | 2.6300ms | 380.2253 Ops/s | 368.3765 Ops/s | |
test_reinforce_speed[reduce-overhead-backward] | 9.8255ms | 8.6970ms | 114.9824 Ops/s | 114.3237 Ops/s | |
test_iql_speed[False-None] | 33.8020ms | 31.9726ms | 31.2768 Ops/s | 31.2965 Ops/s | |
test_iql_speed[False-backward] | 46.6541ms | 44.9193ms | 22.2622 Ops/s | 22.0245 Ops/s | |
test_iql_speed[True-None] | 12.5309ms | 10.6012ms | 94.3288 Ops/s | 93.8794 Ops/s | |
test_iql_speed[True-backward] | 23.6371ms | 21.4401ms | 46.6416 Ops/s | 46.1691 Ops/s | |
test_iql_speed[reduce-overhead-None] | 11.6367ms | 10.8209ms | 92.4141 Ops/s | 93.0866 Ops/s | |
test_iql_speed[reduce-overhead-backward] | 22.6933ms | 21.9945ms | 45.4660 Ops/s | 44.8262 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-ListStorage-RandomSampler-4000] | 7.2728ms | 4.9754ms | 200.9892 Ops/s | 196.1090 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-LazyMemmapStorage-RandomSampler-10000] | 2.8577ms | 0.4842ms | 2.0653 KOps/s | 2.0366 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-LazyTensorStorage-RandomSampler-10000] | 0.7034ms | 0.4615ms | 2.1670 KOps/s | 2.1771 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-4000] | 7.8779ms | 4.7795ms | 209.2275 Ops/s | 218.8328 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-LazyMemmapStorage-SamplerWithoutReplacement-10000] | 1.2849ms | 0.4756ms | 2.1028 KOps/s | 2.0833 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-LazyTensorStorage-SamplerWithoutReplacement-10000] | 0.6555ms | 0.4532ms | 2.2066 KOps/s | 2.1974 KOps/s | |
test_rb_sample[TensorDictReplayBuffer-LazyMemmapStorage-sampler6-10000] | 2.1937ms | 1.5879ms | 629.7817 Ops/s | 624.0680 Ops/s | |
test_rb_sample[TensorDictReplayBuffer-LazyTensorStorage-sampler7-10000] | 2.2815ms | 1.5728ms | 635.8093 Ops/s | 635.6065 Ops/s | |
test_rb_sample[TensorDictPrioritizedReplayBuffer-ListStorage-None-4000] | 5.1475ms | 4.9134ms | 203.5267 Ops/s | 197.6443 Ops/s | |
test_rb_sample[TensorDictPrioritizedReplayBuffer-LazyMemmapStorage-None-10000] | 3.3958ms | 0.6206ms | 1.6113 KOps/s | 1.5831 KOps/s | |
test_rb_sample[TensorDictPrioritizedReplayBuffer-LazyTensorStorage-None-10000] | 0.8586ms | 0.5923ms | 1.6884 KOps/s | 1.6668 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-ListStorage-RandomSampler-4000] | 4.9502ms | 4.7784ms | 209.2766 Ops/s | 210.0597 Ops/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyMemmapStorage-RandomSampler-10000] | 0.9915ms | 0.4872ms | 2.0526 KOps/s | 2.0699 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyTensorStorage-RandomSampler-10000] | 0.6161ms | 0.4589ms | 2.1790 KOps/s | 2.1717 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-4000] | 5.4174ms | 4.7964ms | 208.4876 Ops/s | 215.4677 Ops/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyMemmapStorage-SamplerWithoutReplacement-10000] | 2.6611ms | 0.4800ms | 2.0835 KOps/s | 2.0592 KOps/s | |
test_rb_iterate[TensorDictReplayBuffer-LazyTensorStorage-SamplerWithoutReplacement-10000] | 0.6703ms | 0.4551ms | 2.1975 KOps/s | 2.2333 KOps/s | |
test_rb_iterate[TensorDictPrioritizedReplayBuffer-ListStorage-None-4000] | 5.2719ms | 4.9217ms | 203.1835 Ops/s | 207.4891 Ops/s | |
test_rb_iterate[TensorDictPrioritizedReplayBuffer-LazyMemmapStorage-None-10000] | 3.6037ms | 0.6315ms | 1.5836 KOps/s | 1.5888 KOps/s | |
test_rb_iterate[TensorDictPrioritizedReplayBuffer-LazyTensorStorage-None-10000] | 0.8070ms | 0.5964ms | 1.6766 KOps/s | 1.6792 KOps/s | |
test_rb_populate[TensorDictReplayBuffer-ListStorage-RandomSampler-400] | 7.4593ms | 4.2030ms | 237.9250 Ops/s | 234.8942 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyMemmapStorage-RandomSampler-400] | 6.5249ms | 2.2456ms | 445.3204 Ops/s | 444.5078 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyTensorStorage-RandomSampler-400] | 4.5023ms | 1.2743ms | 784.7262 Ops/s | 641.7431 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-400] | 0.4064s | 12.2541ms | 81.6057 Ops/s | 235.3285 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyMemmapStorage-SamplerWithoutReplacement-400] | 11.6144ms | 2.3791ms | 420.3247 Ops/s | 425.9836 Ops/s | |
test_rb_populate[TensorDictReplayBuffer-LazyTensorStorage-SamplerWithoutReplacement-400] | 1.9675ms | 1.2581ms | 794.8339 Ops/s | 775.8017 Ops/s | |
test_rb_populate[TensorDictPrioritizedReplayBuffer-ListStorage-None-400] | 6.4963ms | 4.2858ms | 233.3265 Ops/s | 225.4994 Ops/s | |
test_rb_populate[TensorDictPrioritizedReplayBuffer-LazyMemmapStorage-None-400] | 5.1001ms | 2.3556ms | 424.5215 Ops/s | 415.0262 Ops/s | |
test_rb_populate[TensorDictPrioritizedReplayBuffer-LazyTensorStorage-None-400] | 6.8785ms | 1.4748ms | 678.0590 Ops/s | 686.5221 Ops/s |
ca359e2
to
0fe3b56
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Labels
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Data
Data-related PR, will launch data-related jobs
Environments
Adds or modifies an environment wrapper
versioning
Versioning change (version number etc)
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.
No description provided.