From b4f6f8b48e022c11af7af2595cadcb17dde8bb92 Mon Sep 17 00:00:00 2001 From: Yifan Zhou Date: Tue, 11 Jun 2024 10:56:16 +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 aef0e70..35723e4 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