PyTorch: Difference between revisions
| Line 15: | Line 15: | ||
import torch | import torch | ||
import torch.nn as nn | import torch.nn as nn | ||
# Training | # Training | ||
| Line 15: | Line 15: | ||
import torch | import torch | ||
import torch.nn as nn | import torch.nn as nn | ||
# Training | # Training | ||