From 6940aaae6bb027fd998d80b752a7685917959bef Mon Sep 17 00:00:00 2001 From: Yifan Zhou Date: Tue, 11 Jun 2024 10:56:27 +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 401138b..bd06b7e 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