From 7c1cf4ebe6dcc4d98a8a010ff849ccb285e372d0 Mon Sep 17 00:00:00 2001 From: Yifan Zhou Date: Thu, 29 Feb 2024 10:31:47 +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 a2b3bad..76a772f 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):