Home
last modified time | relevance | path

Searched refs:train_dataset (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dget_start_xpu.rst238 train_dataset = torchvision.datasets.CIFAR10(
244 train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=128)
297 train_dataset = torchvision.datasets.CIFAR10(
303 train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=128)
H A Drandomness.rst178 train_dataset,
/aosp_15_r20/external/pytorch/functorch/examples/dp_cifar10/
H A Dcifar10_opacus.py165 train_dataset = CIFAR10(
170 train_dataset,
171 batch_size=int(args.sample_rate * len(train_dataset)),
H A Dcifar10_transforms.py229 train_dataset = CIFAR10(
234 train_dataset,
235 batch_size=int(args.sample_rate * len(train_dataset)),
/aosp_15_r20/external/libopus/dnn/torch/neural-pitch/
H A Dtraining.py88 train_dataset, test_dataset = torch.utils.data.random_split(dataset_training, [0.95,0.05], generato… variable
91 train_dataloader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffl…
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/
H A DTensorFlow_Numpy_Distributed_Image_Classification.ipynb145 " train_dataset = tfds.load('mnist', split='train', shuffle_files=True, \n",
152 " images, labels = next(iter(train_dataset.take(1)))\n",
334 " for inputs, labels in train_dataset:\n",
523 " for inputs, labels in train_dataset:\n",
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dintegration.cpp115 auto train_dataset = in test_mnist() local
121 torch::data::make_data_loader(std::move(train_dataset), batch_size); in test_mnist()