更新 'gpu/train.py'

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')