From fcde56995a8d1092dbdf527513a228cb6e689d90 Mon Sep 17 00:00:00 2001 From: Yifan Zhou Date: Fri, 1 Mar 2024 13:40:29 +0800 Subject: [PATCH] Update Conv CNN.py --- QMLS Code/Conv CNN.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/QMLS Code/Conv CNN.py b/QMLS Code/Conv CNN.py index 76a772f..9e784ae 100644 --- a/QMLS Code/Conv CNN.py +++ b/QMLS Code/Conv CNN.py @@ -1,5 +1,5 @@ import torch -import torch.nn as nn +import torch.nn as nn import torch.nn.functional as F class CNN(nn.Module): def __init__(self, n):