标签:utils

PyTorch手册

PyTorch:torch.utils.data

PyTorch:torch.utils.data
class torch.utils.data.Dataset 表示Dataset的抽象类。 所有其他数据集都应该进行子类化。所有子类应该override__len__和__getitem__,前者提供了数据集的大小,后者支持整数索引,范围...

AI君 1年前 (2019-03-28) 117℃ 0喜欢

PyTorch手册

PyTorch:torch.utils.model_zoo

PyTorch:torch.utils.model_zoo
torch.utils.model_zoo.load_url(url, model_dir=None) 在给定URL上加载Torch序列化对象。 如果对象已经存在于 model_dir 中,则将被反序列化并返回。URL的文件名部分应遵循命...

AI君 1年前 (2019-03-28) 128℃ 0喜欢