56
edits
Changes
→A More Intermediate Example
=== A More Intermediate Example ===
Here is a toy model that contains two linear layers. Each linear layer is designed to run a separate GPU[https://pytorch.org/tutorials/intermediate/model_parallel_tutorial.html].
import torch
