Alternatively, you could also use the functional API from torchvision
as given in this example.
1 Like
Alternatively, you could also use the functional API from torchvision
as given in this example.