diff --git a/QMLS Code/Conv CNN.py b/QMLS Code/Conv CNN.py index ba08220..ca8956a 100644 --- a/QMLS Code/Conv CNN.py +++ b/QMLS Code/Conv CNN.py @@ -1,6 +1,6 @@ import torch import torch.nn as nn -import torch.nn.functional as F +import torch.nn.functional as F class CNN(nn.Module): def __init__(self, n): super(CNN, self).__init__()