chore: remove unused code

XuehaiPan · XuehaiPan · commit 92cc6ea38b03 · 2022-09-24T23:33:46.000+08:00
diff --git a/examples/distributed/few-shot/maml_omniglot.py b/examples/distributed/few-shot/maml_omniglot.py
@@ -160,32 +160,6 @@ def main():
         plot(log)
 
 
-def partitioner(net_rref, x_spt, y_spt, x_qry, y_qry, n_inner_iter, task_num):
-    world_size = todist.get_world_size()
-    offset = 1 if task_num < world_size else 0
-
-    partitions = []
-    for task_id in range(task_num):
-        rank = task_id % world_size + offset
-        partitions.append(
-            (
-                rank,
-                (
-                    net_rref,
-                    x_spt[task_id],
-                    y_spt[task_id],
-                    x_qry[task_id],
-                    y_qry[task_id],
-                    n_inner_iter,
-                    task_id,
-                ),
-                None,
-            )
-        )
-
-    return partitions
-
-
 def transpose_mean_reducer(results):
     qry_losses, qry_accs = tuple(zip(*results))
     qry_loss = torch.mean(torch.stack(qry_losses))
