From 7d2e3dbf6fb347a5af6120dccee6303e4995e6b6 Mon Sep 17 00:00:00 2001 From: wjtest1201 Date: Tue, 30 Aug 2022 19:48:10 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'gpu/train.py'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- gpu/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gpu/train.py b/gpu/train.py index f4c09c9..c7075ed 100755 --- a/gpu/train.py +++ b/gpu/train.py @@ -30,6 +30,7 @@ from torch.optim import SGD from torch.utils.data import DataLoader from torchvision.transforms import ToTensor import argparse +import datetime # Training settings parser = argparse.ArgumentParser(description='PyTorch MNIST Example')