Unverified Commit 0b8d0753 authored by vfdev's avatar vfdev Committed by GitHub
Browse files

Remove style (#469)

Update example as style arg was removed in https://github.com/facebookresearch/fairscale/pull/345
parent 80cc7559
...@@ -29,7 +29,6 @@ def run(rank, world_size): ...@@ -29,7 +29,6 @@ def run(rank, world_size):
model = MultiProcessPipe( model = MultiProcessPipe(
model, model,
balance=[2, 1], balance=[2, 1],
style=MultiProcessPipe.MultiProcess,
worker_map={0: "worker0", 1: "worker1"}, # Needed to convert ranks to RPC worker names worker_map={0: "worker0", 1: "worker1"}, # Needed to convert ranks to RPC worker names
input_device=device, input_device=device,
).to(device) ).to(device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment