PyTorch: Difference between revisions

Line 15: Line 15:
import torch
import torch
import torch.nn as nn
import torch.nn as nn


# Training
# Training