Можно ли сохранить объект PyTorch dataLoader в файл или что-то в этом роде?

Я работаю над проектом машинного обучения, и набор данных, который я использую, довольно велик. Каждый раз, когда я запускаю свою функцию getLoaders (которая просто вызывает Imagefolder(), разделяет данные и вызывает torch.utils.data.DataLoader(), у меня уходит более 20 минут)

      loader = torch.utils.data.DataLoader(test_pair_list, batch_size=batchSize, sampler=valSampler)

Итак, есть ли способ сохранить эти объекты в файле, чтобы getLoader вызывался только один раз? Я просмотрел документацию PyTorch, но мне не повезло.

0 ответов

Другие вопросы по тегам