
pytorch
文章平均质量分 63
fightingxyz
这个作者很懒,什么都没留下…
展开
-
Pytorch-Resnet迁移学习记录
暂时不解释,仅作为记录:import torchfrom torchvision import datasets, models, transformsimport torch.nn as nnimport torch.optim as optimfrom torch.utils.data import DataLoaderimport timeimport numpy as npimport matplotlib.pyplot as pltimport osimage_tran原创 2021-07-14 10:28:03 · 544 阅读 · 0 评论 -
ValueError: num_samples should be a positive integer value, but got num_samp=0
dataloader = DataLoader( ImageDataset("./data/%s" % opt.dataset_name, transforms_=transforms_), batch_size=opt.batch_size, shuffle=True, num_workers=opt.n_cpu, )使用上诉数据设置出错。原因:shuffle的参数设置错误导致。解决方法:因为已经有batch_...原创 2021-06-02 16:21:45 · 16568 阅读 · 13 评论