模式识别与机器学习课程作业,根据网上代码自己搞了一个,用pytorch写的
参考https://blog.csdn.net/qq_27492735/article/details/95318308
import torch
import torch.nn as nn
import torchvision
import torchvision.transforms as transforms
from torch.utils.data import Dataset
from torch.autograd import Variable
import numpy as np
import matplotlib.pyplot as plt
def setup_seed(seed):
torch.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
np.random.seed(seed)
#random.seed(seed)#没有用到这个库
torch.backends.cudnn.deterministic = True
setup_seed(2)
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
if torch.cuda.is_available():
print("using GPU")
else:
print("using CPU")
## Some super parameters
batch_size = 64
num_epochs = 1
input_layer_size = 784
learning_rate = 0.01
train_ornot = False