correcting the pipeline rpc example (#773)

Co-authored-by: Aleksandr Panchul (CSI Interfusion Inc) <[email protected]>

Author: panchul

.gitignore

@@ -12,3 +12,6 @@ snli/results
 word_language_model/model.pt
 fast_neural_style/saved_models
 fast_neural_style/saved_models.zip
+
+# vi backups
+*~

distributed/rpc/pipeline/main.py

@@ -226,7 +226,8 @@ def parameter_rrefs(self):
 image_h = 128
 
 
-def run_master(num_split):
+def run_master(split_size):
+
     # put the two model parts on worker1 and worker2 respectively
     model = DistResNet50(split_size, ["worker1", "worker2"])
     loss_fn = nn.MSELoss()
@@ -285,4 +286,4 @@ def run_worker(rank, world_size, num_split):
         tik = time.time()
         mp.spawn(run_worker, args=(world_size, num_split), nprocs=world_size, join=True)
         tok = time.time()
-        print(f"number of splits = {num_split}, execution time = {tok - tik}")
+        print(f"number of splits = {num_split}, execution time = {tok - tik}")