经典的28*28手写数据集
电脑配置是3070tilaptop和12代i7
在nivdia-msi中查看,同样是个别时候才有高占用率
batch_size设置为64
DataLoader中num_works设置为8,pin_memory设置为truel
神经网络有三个隐层,神经元依次是300 200 100



经典的28*28手写数据集
电脑配置是3070tilaptop和12代i7
在nivdia-msi中查看,同样是个别时候才有高占用率
batch_size设置为64
DataLoader中num_works设置为8,pin_memory设置为truel
神经网络有三个隐层,神经元依次是300 200 100



mnist数据集太小了,所以gpu用不满,这是很正常的