Searched refs:create_mnist_dataset (Results 1 – 3 of 3) sorted by relevance
31 from tests.st.summary.dataset import create_mnist_dataset118 ds_train = create_mnist_dataset("train", num_samples=num_samples)121 ds_eval = create_mnist_dataset("test")
30 from tests.st.summary.dataset import create_mnist_dataset100 ds_train = create_mnist_dataset('train', num_samples=1, batch_size=1)
25 def create_mnist_dataset(mode='train', num_samples=2, batch_size=2): function