Skip to content
Browse files

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

Signed-off-by: terrytangyuan <>
  • 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
@@ -260,6 +260,7 @@ Example (also see a full training [example](examples/

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.