Skip to content
Permalink
Browse files

Add missing import for torch.nn.functional (#1048)

Signed-off-by: terrytangyuan <terrytangyuan@gmail.com>
  • Loading branch information...
terrytangyuan authored and alsrgv committed May 7, 2019
1 parent c9fa3f7 commit 73d860f2396321761e0f5ef6fe934130afd69094
Showing with 1 addition and 0 deletions.
  1. +1 −0 README.md
@@ -260,6 +260,7 @@ Example (also see a full training [example](examples/pytorch_mnist.py)):

```python
import torch
import torch.nn.functional as F
import horovod.torch as hvd
# Initialize Horovod

0 comments on commit 73d860f

Please sign in to comment.
You can’t perform that action at this time.