Torch.nn.functional or torch.nn

vision
2

Have a look at this post for some explanations. :slight_smile:

1 Like