pytorch/test
Rohan Varma c0ebeca1a8 [Functional Optim] Test kwargs parity for SGD (#62078)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62078

Ensure that kwarg arguments such as momentum and weight decay maintain
parity between optimizer.step and step_param.
ghstack-source-id: 134330377

Test Plan: CI

Reviewed By: SciPioneer

Differential Revision: D29837942

fbshipit-source-id: 1ae39648fc26aebd8aaef1a7ac0e03b598a8ed60
2021-07-26 22:11:40 -07:00
..
ao/sparsity [pruner] add Conv2d support (#61778) 2021-07-22 23:00:31 -07:00
backward_compatibility [quant] Remove calls to .item() for fake_quant_on (#61921) 2021-07-21 10:13:06 -07:00
benchmark_utils Make torch.utils.bencmark numpy free (#60564) 2021-06-30 14:17:32 -07:00
bottleneck_test
cpp [PyTorch][Edge] Serializing sub modules with same names (#61933) 2021-07-26 16:31:48 -07:00
cpp_api_parity ENH Adds nn.ReflectionPad3d (#59791) 2021-06-21 10:53:14 -07:00
cpp_extensions Add a test case for findDanglingImpls (#61104) 2021-07-07 13:34:16 -07:00
custom_backend [Pytorch backend delegation] Preprocess to accept (#58873) 2021-06-11 10:16:00 -07:00
custom_operator
distributed [Reland] [DDP] Implement a hook which performs FunctionalSGD step. (#62177) 2021-07-26 11:56:56 -07:00
distributions Improve error message on invalid values to Distribution methods (#61056) 2021-07-06 15:44:55 -07:00
error_messages
expect [jit] Set debug name for value coming out of GetAttr nodes. (#59123) 2021-06-09 12:24:55 -07:00
fx flatten operation (resnet50) (#61265) 2021-07-16 16:06:10 -07:00
jit [PyTorch][Edge] Serializing sub modules with same names (#61933) 2021-07-26 16:31:48 -07:00
jit_hooks
mobile [PyTorch][Edge] Tests for QuantizationFx API on lite modules (#60476) 2021-07-08 10:40:08 -07:00
onnx [ONNX] add supplement for standardOps low precision cast (#60731) (#61561) 2021-07-21 15:10:36 -07:00
optim To add Rectified Adam Algorithm to Optimizers (#58968) 2021-06-23 18:27:57 -07:00
package [package] merge test_torchscript into test_package_script (#61807) 2021-07-19 18:23:45 -07:00
quantization [bc-breaking][quant][graphmode][fx] Produce dequant - fp_op - quant pattern for copy nodes (#61763) 2021-07-25 19:49:13 -07:00
scripts
test_img
typing
HowToWriteTestsUsingFileCheck.md Allow for heterogenous List and Dict values + Improve container typing algorithm (#57137) 2021-07-10 14:29:05 -07:00
linear.py
run_test.py Parity tests for functional optimizer step_param (#61756) 2021-07-26 19:03:22 -07:00
simulate_nccl_errors.py
test_ao_sparsity.py [sparsity] Lambda Scheduler (#59771) 2021-07-02 21:39:38 -07:00
test_autocast.py
test_autograd.py Add default Saved Variable hooks (#61834) 2021-07-26 08:14:32 -07:00
test_binary_ufuncs.py OpInfo: bitwise_and (#61349) 2021-07-22 07:04:17 -07:00
test_buffer_protocol.py Implement NumPy-like frombuffer tensor constructor. (#59077) 2021-07-23 13:17:48 -07:00
test_bundled_images.py
test_bundled_inputs.py
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py
test_cpp_extensions_jit.py Adding super calls to JIT test case setUp and tearDown (#61922) 2021-07-20 15:08:44 -07:00
test_cuda.py Makes a streaming backward test try gradient stealing more directly (#60065) 2021-07-19 20:39:55 -07:00
test_cuda_primary_ctx.py
test_dataloader.py fix for #60319 , forcing to use fork as start method in test/test_dat… (#60868) 2021-06-29 09:30:37 -07:00
test_datapipe.py Sort imports of test_datapipe.py (#61312) 2021-07-12 15:33:20 -07:00
test_determination.py Disable group group backend rpc tests from running on CI (#60407) 2021-06-23 10:58:31 -07:00
test_dispatch.py Add a test case for findDanglingImpls (#61104) 2021-07-07 13:34:16 -07:00
test_foreach.py Foreach Test Refactor: Pointwise, Min/Max-imum (#61327) 2021-07-21 21:59:57 -07:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py [Functional Optim] Test kwargs parity for SGD (#62078) 2021-07-26 22:11:40 -07:00
test_futures.py remove unused type: ignore directives (#60006) 2021-06-18 07:23:31 -07:00
test_fx.py [1/n] Update testing lib*.so path (#61960) 2021-07-24 05:16:35 -07:00
test_fx_experimental.py Alias for polygamma (#59691) 2021-07-16 00:06:27 -07:00
test_gen_backend_stubs.py
test_import_time.py First step to rearrange files in tools folder (#60473) 2021-06-24 10:13:58 -07:00
test_indexing.py [bc-breaking] Dispatch index_put with boolean mask argument to masked_fill (#61612) 2021-07-19 22:53:14 -07:00
test_jit.py Back out "Revert D29687143: [3/N] Nnapi Backend Delegate Preprocess: Basic OSS Test" (#61878) 2021-07-21 11:58:45 -07:00
test_jit_cuda_fuser.py
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py [nnc] Get rid of fuser trigger counters (#57334) 2021-06-29 22:22:15 -07:00
test_jit_legacy.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_kernel_launch_checks.py Paren-matching kernel launch check without external deps (#60778) 2021-06-28 10:18:04 -07:00
test_license.py
test_linalg.py fix mm not correctly report TORCH_CHECK failure issue (#61394) 2021-07-12 12:50:51 -07:00
test_logging.py
test_metal.py
test_mkldnn.py enable check trace when tracing a mkldnn model (#61241) 2021-07-19 11:03:53 -07:00
test_mobile_optimizer.py
test_model_dump.py model_dump: Fix non-counting and double-counting bugs in tensor memory (#60702) 2021-07-10 15:16:34 -07:00
test_module_init.py Adds _LazyInstanceNorm and LazyInstanceNormXd (#60982) 2021-07-21 06:45:45 -07:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py Stop warning on .names() access in max_pool2d and max_pool2d_backward (#60059) 2021-06-17 10:34:41 -07:00
test_namedtuple_return_api.py [quant] Remove calls to .item() for fake_quant_on (#61921) 2021-07-21 10:13:06 -07:00
test_native_functions.py
test_nn.py ENH Adds no batch dim support for AdativeMaxPool*D (#61847) 2021-07-26 07:35:36 -07:00
test_nnapi.py Handle simple NNAPI flatten NHWC case (#61796) 2021-07-26 10:59:04 -07:00
test_numba_integration.py
test_numpy_interop.py Warn about poor performance creating Tensor from list of numpy.array's (#51680) 2021-07-22 12:02:50 -07:00
test_openmp.py
test_ops.py Add complex64 dtype for OpInfo Reference testing (#61627) 2021-07-15 13:40:37 -07:00
test_optim.py Implement and benchmark a torch.optim.multi_tensor.adagrad implementation (#59155) 2021-07-07 08:08:32 -07:00
test_overrides.py [skip ci] Fix "arugment" typos (#61459) 2021-07-15 15:20:18 -07:00
test_package.py
test_profiler.py
test_pruning_op.py
test_public_bindings.py Fix _C public bindings test (#61088) 2021-07-21 11:50:37 -07:00
test_python_dispatch.py Remove default arguments before calling to __torch_dispatch__ (#61123) 2021-07-23 10:41:35 -07:00
test_pytree.py
test_quantization.py [quant] Create FusedMovingAvgObsFakeQuantize for QAT (#61691) 2021-07-21 10:13:04 -07:00
test_reductions.py Use cascade-summation to improve nansum accuracy (#61082) 2021-07-19 21:47:43 -07:00
test_segment_reductions.py [torch][segment_reduce] Update default values when initial value is not set (#61266) 2021-07-07 13:34:10 -07:00
test_serialization.py Hacky support for meta tensor serialization. (#62192) 2021-07-26 14:33:45 -07:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py [Testing] Adding reference tests to OpInfo class (#59369) 2021-06-23 19:26:08 -07:00
test_show_pickle.py
test_sort_and_select.py add BFloat16 support for topk on CPU (#59547) 2021-07-19 16:06:24 -07:00
test_sparse.py Remove torch._bmm and remove torch.bmm deterministic arg documentation (#61629) 2021-07-20 10:55:43 -07:00
test_sparse_csr.py use torch.bucketize into_sparse_csr implementation (+ additional tests) (#61340) 2021-07-20 15:44:25 -07:00
test_spectral_ops.py [ROCM] fix bug in #60313 (#61073) 2021-07-13 07:08:17 -07:00
test_static_runtime.py [Static Runtime] Support prim::GetAttr/SetAttr (#61505) 2021-07-10 14:06:06 -07:00
test_tensor_creation_ops.py add BFloat16 operators on CPU: arange, acosh, asinh, atanh, exp2, digamma, trigamma, polygamma (#60444) 2021-07-22 08:13:22 -07:00
test_tensorboard.py
test_tensorexpr.py [TensorExpr] Do not fuse float16 values. (#61569) 2021-07-14 12:53:59 -07:00
test_tensorexpr_pybind.py [nnc] Insert alloc/free at global scope (#61725) 2021-07-16 08:42:24 -07:00
test_testing.py Enable torch.isclose to suppport bool tensors (#61271) 2021-07-21 18:50:14 -07:00
test_throughput_benchmark.py
test_torch.py [ROCm] Skip test_masked_scatter_large_tensor_cuda (#61313) 2021-07-09 10:27:08 -07:00
test_type_hints.py
test_type_info.py
test_type_promotion.py
test_typing.py
test_unary_ufuncs.py Fix test failures with some glibc libraries (#60450) 2021-06-23 07:49:27 -07:00
test_utils.py Fix breakpad build + add test canary (#60990) 2021-07-06 14:15:07 -07:00
test_view_ops.py Throw RuntimeError when numpy() is called on a tensor with conjugate or negative bit set (#61925) 2021-07-23 11:28:36 -07:00
test_vmap.py Improve testing of inplace views (#59891) 2021-06-22 12:28:09 -07:00
test_vulkan.py
test_xnnpack_integration.py