[fix] examples: fix naming style of helper functions (#334)

examples/helpers.py

@@ -10,13 +10,13 @@ def dist_init(rank, world_size):
     dist.init_process_group(backend=backend, init_method="tcp://localhost:29501", rank=rank, world_size=world_size)
 
 
-def getModel():
+def get_model():
     return nn.Sequential(torch.nn.Linear(10, 10), torch.nn.ReLU(), torch.nn.Linear(10, 5))
 
 
-def getData(n_batches=1):
+def get_data(n_batches=1):
     return [(torch.randn(20, 10), torch.randint(0, 2, size=(20, 1)).squeeze()) for i in range(n_batches)]
 
 
-def getLossFun():
+def get_loss_fun():
     return F.nll_loss

examples/tutorial_oss.py

 import time
 from typing import Optional, Union, cast
 
-from helpers import dist_init, getData, getLossFun, getModel
+from helpers import dist_init, get_data, get_loss_fun, get_model
 import torch
 import torch.distributed as dist
 import torch.multiprocessing as mp
@@ -21,9 +21,9 @@ def train(rank: int, world_size: int, epochs: int, use_oss: bool):
     device = torch.device("cpu") if DEVICE == "cpu" else rank  # type:ignore
 
     # Problem statement
-    model = getModel().to(device)
-    dataloader = getData(n_batches=1)
-    loss_fn = getLossFun()
+    model = get_model().to(device)
+    dataloader = get_data(n_batches=1)
+    loss_fn = get_loss_fun()
 
     optimizer: Optional[Union[OSS, torch.optim.SGD]] = None
 

examples/tutorial_pipe.py

-from helpers import getData, getLossFun, getModel
+from helpers import get_data, get_loss_fun, get_model
 import torch
 import torch.optim as optim
 
@@ -7,9 +7,9 @@ import fairscale
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 RANK = 0  # example
 
-model = getModel()
-data, target = getData()[0]
-loss_fn = getLossFun()
+model = get_model()
+data, target = get_data()[0]
+loss_fn = get_loss_fun()
 
 model = fairscale.nn.Pipe(model, balance=[2, 1])
 

examples/tutorial_pipe_multiprocess.py

 import os
 
-from helpers import dist_init, getData, getLossFun, getModel
+from helpers import dist_init, get_data, get_loss_fun, get_model
 import torch
 import torch.distributed as dist
 import torch.multiprocessing as mp
@@ -20,9 +20,9 @@ def run(rank, world_size):
     dist.rpc.init_rpc(f"worker{rank}", rank=rank, world_size=world_size)
     initialize_model_parallel(1, world_size)
 
-    model = getModel()
-    data, target = getData()[0]
-    loss_fn = getLossFun()
+    model = get_model()
+    data, target = get_data()[0]
+    loss_fn = get_loss_fun()
 
     device = torch.device("cuda", rank) if DEVICE == "cuda" else torch.device("cpu")
 

examples/tutorial_pipe_rpc.py

@@ -3,7 +3,7 @@
 
 import os
 
-from helpers import dist_init, getData, getLossFun, getModel
+from helpers import dist_init, get_data, get_loss_fun, get_model
 import torch
 import torch.optim as optim
 import torch_pg
@@ -37,9 +37,9 @@ def run(rank, world_size):
         torch.distributed.rpc.shutdown()
         return
 
-    model = getModel()
-    data, target = getData()[0]
-    loss_fn = getLossFun()
+    model = get_model()
+    data, target = get_data()[0]
+    loss_fn = get_loss_fun()
 
     device = torch.device("cuda", rank)