
pytorch
Tinet-
AI探道夫一枚
展开
-
Pytorch在一个Batch内加载两个大小不等的Dataset
在使用Pytorch进行训练时,有时需要在一个batch同时加载两个不同的数据集。考虑到两个数据集大小可能不同,遍历两个数据及可以按如下操作:dataloaders1 = DataLoader(DummyDataset(0, 100), batch_size=10, shuffle=True)dataloaders2 = DataLoader(DummyDataset(0, 200), batch_size=10, shuffle=True)num_epochs = 10for epoch in原创 2022-04-10 15:36:32 · 3126 阅读 · 0 评论 -
安装pytorch因网速问题失败解决方案
前言每次从官网提供的下载方式,使用conda安装pytorch的时候都会因为网速问题导致失败。使用conda清华源安装也不失为一种解决方案,然而这种方式在指定cuda版本时非常不方便。本文主要针对这一问题给出解决方案。方法以安装pytorh-nightly, cuda80, linux系统为例。从pytorch官网选择 pip安装,并选定对应的系统, cuda版本, python版本。这时...原创 2019-04-30 14:45:40 · 3157 阅读 · 1 评论