摘要：
构建数据集： 网络搭建： class nn(nn.Module): def __init__(self): super(nn, self).__init__() self.nets = nn.Sequential( # 网络细节 # 针对单元（图像···） ) def forward(self, x 阅读全文
posted @ 2019-11-15 20:50
Junzhao
阅读(204)
评论(0)
推荐(0)
摘要：
num_workers通过影响数据加载速度，从而影响训练速度 在调整num_workers的过程中，发现训练速度并没有变化 原因在于： num_workers是加载数据（batch）的线程数目 当加载batch的时间 < 数据训练的时间 GPU每次训练完都可以直接从CPU中取到next batch的 阅读全文
posted @ 2019-11-15 20:00
Junzhao
阅读(17567)
评论(0)
推荐(1)
摘要：
保存数据预处理操作，在训练时可直接读取，节省预处理时间 import os ... if not os.path.exists('filename'): # 预处理 np.savez('filename.npz',x_array=x_array,y_array=y_array) else: file 阅读全文
posted @ 2019-11-15 16:00
Junzhao
阅读(627)
评论(0)
推荐(0)
浙公网安备 33010602011771号