Back out "Replace LR access with wrapper"

Summary:
X-link: pytorch/FBGEMM#3857

X-link: facebookresearch/FBGEMM#949

Original commit changeset: e175eecac4c5

Original Phabricator Diff: D71444136

Reviewed By: sryap

Differential Revision: D71578251

Author: spcyppt

torchrec/distributed/batched_embedding_kernel.py

@@ -190,7 +190,7 @@ def __init__(
         state: Dict[Any, Any] = {}
         param_group: Dict[str, Any] = {
             "params": [],
-            "lr": emb_module.get_learning_rate(),
+            "lr": emb_module.optimizer_args.learning_rate,
         }
 
         params: Dict[str, Union[torch.Tensor, ShardedTensor]] = {}
@@ -383,7 +383,7 @@ def get_optimizer_pointwise_shard_metadata_and_global_metadata(
         state: Dict[Any, Any] = {}
         param_group: Dict[str, Any] = {
             "params": [],
-            "lr": emb_module.get_learning_rate(),
+            "lr": emb_module.optimizer_args.learning_rate,
         }
 
         params: Dict[str, Union[torch.Tensor, ShardedTensor]] = {}

torchrec/modules/fused_embedding_modules.py

@@ -68,7 +68,7 @@ def __init__(  # noqa C901
         state: Dict[Any, Any] = {}
         param_group: Dict[str, Any] = {
             "params": [],
-            "lr": emb_module.get_learning_rate(),
+            "lr": emb_module.optimizer_args.learning_rate,
         }
 
         params: Dict[str, torch.Tensor] = {}