diff --git a/QMLS Code/Conv CNN.py b/QMLS Code/Conv CNN.py index 9e784ae..c259ba3 100644 --- a/QMLS Code/Conv CNN.py +++ b/QMLS Code/Conv CNN.py @@ -1,4 +1,4 @@ -import torch +import torch import torch.nn as nn import torch.nn.functional as F class CNN(nn.Module):