From 17aeedfe2ec3c0375dcb0cec6c634e6e5e083809 Mon Sep 17 00:00:00 2001 From: Yifan Zhou Date: Tue, 11 Jun 2024 10:55:42 +0800 Subject: [PATCH] Update Task 1.py --- QMLS Code/Task 1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/QMLS Code/Task 1.py b/QMLS Code/Task 1.py index 1ada488..948cc70 100644 --- a/QMLS Code/Task 1.py +++ b/QMLS Code/Task 1.py @@ -1,5 +1,5 @@ import torch -import torch.nn as nn +import torch.nn as nn import torch.optim as optim import torchvision import torchvision.transforms as transforms